├── OIR-Bench └── inform.txt ├── README.md ├── assets ├── OIR_result_1.png ├── OIR_result_2.png ├── OIR_result_3.png ├── search_metric_1.png └── search_metric_2.png ├── configs ├── basic_config.py ├── multi_object_edit.yaml └── single_object_edit.yaml ├── models ├── attention.py ├── attention_processor.py ├── autoencoder_kl.py ├── mask_clip.py ├── transformer_2d.py ├── unet_2d_blocks.py ├── unet_2d_condition.py └── vae.py ├── multi_object_edit.py ├── requirements.txt ├── sampler ├── ddim_inversion.py └── ddim_scheduling.py ├── single_object_edit.py └── utils ├── basic_utils.py ├── candidate_images_generation.py ├── oir.py └── optimal_candidate_selection.py /OIR-Bench/inform.txt: -------------------------------------------------------------------------------- 1 | Download OIR-Bench and put it here. -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/README.md -------------------------------------------------------------------------------- /assets/OIR_result_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/assets/OIR_result_1.png -------------------------------------------------------------------------------- /assets/OIR_result_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/assets/OIR_result_2.png -------------------------------------------------------------------------------- /assets/OIR_result_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/assets/OIR_result_3.png -------------------------------------------------------------------------------- /assets/search_metric_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/assets/search_metric_1.png -------------------------------------------------------------------------------- /assets/search_metric_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/assets/search_metric_2.png -------------------------------------------------------------------------------- /configs/basic_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/configs/basic_config.py -------------------------------------------------------------------------------- /configs/multi_object_edit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/configs/multi_object_edit.yaml -------------------------------------------------------------------------------- /configs/single_object_edit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/configs/single_object_edit.yaml -------------------------------------------------------------------------------- /models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/models/attention.py -------------------------------------------------------------------------------- /models/attention_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/models/attention_processor.py -------------------------------------------------------------------------------- /models/autoencoder_kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/models/autoencoder_kl.py -------------------------------------------------------------------------------- /models/mask_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/models/mask_clip.py -------------------------------------------------------------------------------- /models/transformer_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/models/transformer_2d.py -------------------------------------------------------------------------------- /models/unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/models/unet_2d_blocks.py -------------------------------------------------------------------------------- /models/unet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/models/unet_2d_condition.py -------------------------------------------------------------------------------- /models/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/models/vae.py -------------------------------------------------------------------------------- /multi_object_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/multi_object_edit.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/requirements.txt -------------------------------------------------------------------------------- /sampler/ddim_inversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/sampler/ddim_inversion.py -------------------------------------------------------------------------------- /sampler/ddim_scheduling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/sampler/ddim_scheduling.py -------------------------------------------------------------------------------- /single_object_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/single_object_edit.py -------------------------------------------------------------------------------- /utils/basic_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/utils/basic_utils.py -------------------------------------------------------------------------------- /utils/candidate_images_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/utils/candidate_images_generation.py -------------------------------------------------------------------------------- /utils/oir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/utils/oir.py -------------------------------------------------------------------------------- /utils/optimal_candidate_selection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/OIR/HEAD/utils/optimal_candidate_selection.py --------------------------------------------------------------------------------