Yinwei Wu Xingyi Yang Xinchao Wang†
LV-Lab, National University of Singapore
†corresponding author
CVPR 2024
git clone https://github.com/WUyinwei-hah/RRNet.git
cd RRNet
pip install -r requirements.txt
Install the dgl from https://dgl.ai/pages/start.html
There are some examples in the ./Dataset, which you can try directly.
You can also download the whole RR dataset from https://drive.google.com/file/d/1rKmeR1k4S6Im8YMq-FSBGK11nNwQzhwM/view?usp=sharing
python3 train.py --data_folder ./Dataset/contain/ --device 0 The generation result and the model checkpoints will be placed at ./generation_result.
python3 generate.py --data_folder ./Dataset/contain/ --save_folder ./generation_result --device 0 @misc{wu2024relation,
title={Relation Rectification in Diffusion Model},
author={Yinwei Wu and Xingyi Yang and Xinchao Wang},
year={2024},
eprint={2403.20249},
archivePrefix={arXiv},
primaryClass={cs.CV}
}