Relation Rectification in Diffusion Model

Yinwei Wu Xingyi Yang Xinchao Wang^†

LV-Lab, National University of Singapore  ^†corresponding author

CVPR 2024

Overview

Installation

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

Data preparation

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

Training

python3 train.py --data_folder ./Dataset/contain/ --device 0 

The generation result and the model checkpoints will be placed at ./generation_result.

Generation

python3 generate.py --data_folder ./Dataset/contain/ --save_folder ./generation_result  --device 0 

Citation

@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}
}