├── .gitignore ├── LICENSE ├── README.md ├── app.py ├── assets ├── ARGS_README.md └── teaser.png ├── dataset ├── anime.png ├── bald_man.jpg ├── bald_man_mask.png ├── bear.png ├── biden.png ├── bird.png ├── bread.png ├── butterfly.jpg ├── car4.jpg ├── cartoon_cat.png ├── dog.png ├── dog2.png ├── dolphin.jpg ├── duck.jpg ├── fatty-corgi.jpg ├── fatty-corgi2.jpg ├── flower2.jpg ├── gcat.jpg ├── girls.jpg ├── grey_cat2.jpg ├── korra.png ├── man3.jpg ├── mercedes.jpg ├── pearl.png ├── shepperd.png ├── smiling_man2.png ├── spiderman.jpg ├── stones.jpeg ├── teddy4.png ├── white_cat.png └── yellow_cat.jpg ├── main.py ├── model ├── __init__.py ├── directional_attentions.py ├── modules │ ├── __pycache__ │ │ ├── dift_utils.cpython-310.pyc │ │ ├── freq_filters.cpython-310.pyc │ │ ├── new_object_detection.cpython-310.pyc │ │ └── register_attention.cpython-310.pyc │ ├── dift_utils.py │ ├── freq_filters.py │ ├── new_object_detection.py │ └── register_attention.py ├── pipeline_sdxl.py └── unet_sdxl.py ├── requirements.txt ├── scripts ├── config.sh ├── edit.sh ├── prompts │ └── p.json └── steps_mode │ ├── 10_steps.sh │ ├── 4_steps.sh │ └── 5_steps.sh ├── src ├── __init__.py ├── ddpm_inversion.py └── ddpm_step.py ├── utils ├── __init__.py ├── args.py ├── pipeline_utils.py └── utils.py └── visualization ├── __init__.py ├── draw_box.py ├── draw_mask.py └── image_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/README.md -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/app.py -------------------------------------------------------------------------------- /assets/ARGS_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/assets/ARGS_README.md -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /dataset/anime.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/anime.png -------------------------------------------------------------------------------- /dataset/bald_man.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/bald_man.jpg -------------------------------------------------------------------------------- /dataset/bald_man_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/bald_man_mask.png -------------------------------------------------------------------------------- /dataset/bear.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/bear.png -------------------------------------------------------------------------------- /dataset/biden.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/biden.png -------------------------------------------------------------------------------- /dataset/bird.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/bird.png -------------------------------------------------------------------------------- /dataset/bread.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/bread.png -------------------------------------------------------------------------------- /dataset/butterfly.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/butterfly.jpg -------------------------------------------------------------------------------- /dataset/car4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/car4.jpg -------------------------------------------------------------------------------- /dataset/cartoon_cat.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/cartoon_cat.png -------------------------------------------------------------------------------- /dataset/dog.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/dog.png -------------------------------------------------------------------------------- /dataset/dog2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/dog2.png -------------------------------------------------------------------------------- /dataset/dolphin.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/dolphin.jpg -------------------------------------------------------------------------------- /dataset/duck.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/duck.jpg -------------------------------------------------------------------------------- /dataset/fatty-corgi.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/fatty-corgi.jpg -------------------------------------------------------------------------------- /dataset/fatty-corgi2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/fatty-corgi2.jpg -------------------------------------------------------------------------------- /dataset/flower2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/flower2.jpg -------------------------------------------------------------------------------- /dataset/gcat.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/gcat.jpg -------------------------------------------------------------------------------- /dataset/girls.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/girls.jpg -------------------------------------------------------------------------------- /dataset/grey_cat2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/grey_cat2.jpg -------------------------------------------------------------------------------- /dataset/korra.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/korra.png -------------------------------------------------------------------------------- /dataset/man3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/man3.jpg -------------------------------------------------------------------------------- /dataset/mercedes.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/mercedes.jpg -------------------------------------------------------------------------------- /dataset/pearl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/pearl.png -------------------------------------------------------------------------------- /dataset/shepperd.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/shepperd.png -------------------------------------------------------------------------------- /dataset/smiling_man2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/smiling_man2.png -------------------------------------------------------------------------------- /dataset/spiderman.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/spiderman.jpg -------------------------------------------------------------------------------- /dataset/stones.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/stones.jpeg -------------------------------------------------------------------------------- /dataset/teddy4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/teddy4.png -------------------------------------------------------------------------------- /dataset/white_cat.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/white_cat.png -------------------------------------------------------------------------------- /dataset/yellow_cat.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/dataset/yellow_cat.jpg -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/main.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/directional_attentions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/model/directional_attentions.py -------------------------------------------------------------------------------- /model/modules/__pycache__/dift_utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/model/modules/__pycache__/dift_utils.cpython-310.pyc -------------------------------------------------------------------------------- /model/modules/__pycache__/freq_filters.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/model/modules/__pycache__/freq_filters.cpython-310.pyc -------------------------------------------------------------------------------- /model/modules/__pycache__/new_object_detection.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/model/modules/__pycache__/new_object_detection.cpython-310.pyc -------------------------------------------------------------------------------- /model/modules/__pycache__/register_attention.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/model/modules/__pycache__/register_attention.cpython-310.pyc -------------------------------------------------------------------------------- /model/modules/dift_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/model/modules/dift_utils.py -------------------------------------------------------------------------------- /model/modules/freq_filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/model/modules/freq_filters.py -------------------------------------------------------------------------------- /model/modules/new_object_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/model/modules/new_object_detection.py -------------------------------------------------------------------------------- /model/modules/register_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/model/modules/register_attention.py -------------------------------------------------------------------------------- /model/pipeline_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/model/pipeline_sdxl.py -------------------------------------------------------------------------------- /model/unet_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/model/unet_sdxl.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/config.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/scripts/config.sh -------------------------------------------------------------------------------- /scripts/edit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/scripts/edit.sh -------------------------------------------------------------------------------- /scripts/prompts/p.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/scripts/prompts/p.json -------------------------------------------------------------------------------- /scripts/steps_mode/10_steps.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/scripts/steps_mode/10_steps.sh -------------------------------------------------------------------------------- /scripts/steps_mode/4_steps.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/scripts/steps_mode/4_steps.sh -------------------------------------------------------------------------------- /scripts/steps_mode/5_steps.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/scripts/steps_mode/5_steps.sh -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/src/__init__.py -------------------------------------------------------------------------------- /src/ddpm_inversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/src/ddpm_inversion.py -------------------------------------------------------------------------------- /src/ddpm_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/src/ddpm_step.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/utils/args.py -------------------------------------------------------------------------------- /utils/pipeline_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/utils/pipeline_utils.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/utils/utils.py -------------------------------------------------------------------------------- /visualization/__init__.py: -------------------------------------------------------------------------------- 1 | from .image_utils import save_results -------------------------------------------------------------------------------- /visualization/draw_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/visualization/draw_box.py -------------------------------------------------------------------------------- /visualization/draw_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/visualization/draw_mask.py -------------------------------------------------------------------------------- /visualization/image_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alimohammadiamirhossein/cora/HEAD/visualization/image_utils.py --------------------------------------------------------------------------------