├── .gitignore ├── README.md ├── dataset ├── CoCoCount.json ├── create_data_CoCoCount.py ├── create_data_compbench.py └── data_compbench.json ├── evaluation_requirements.txt ├── evaluation_script.py ├── figures └── teaser.jpg ├── pipeline ├── __init__.py ├── attention_processors.py ├── attention_store_counting.py ├── countgen.ipynb ├── mask_extraction │ ├── dbscan_mask_extract.py │ ├── extract_mask.py │ ├── postprocess.py │ ├── relayout.py │ └── utils_masks.py ├── pipeline_config.yaml ├── run_countgen.py └── self_counting_sdxl_pipeline.py ├── requirements.txt ├── train_relayout ├── __init__.py ├── data_creation │ ├── __init__.py │ ├── generate_counting_unet_data.py │ └── matching_algorithm.py ├── dataloader.py ├── loss.py ├── relayout_requirements.txt ├── train_unet.py ├── unet_config.yaml └── utils.py └── utils ├── __init__.py ├── attention_utils.py ├── counting_words_extract.py ├── generate_random_masks.py └── loss_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/README.md -------------------------------------------------------------------------------- /dataset/CoCoCount.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/dataset/CoCoCount.json -------------------------------------------------------------------------------- /dataset/create_data_CoCoCount.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/dataset/create_data_CoCoCount.py -------------------------------------------------------------------------------- /dataset/create_data_compbench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/dataset/create_data_compbench.py -------------------------------------------------------------------------------- /dataset/data_compbench.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/dataset/data_compbench.json -------------------------------------------------------------------------------- /evaluation_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/evaluation_requirements.txt -------------------------------------------------------------------------------- /evaluation_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/evaluation_script.py -------------------------------------------------------------------------------- /figures/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/figures/teaser.jpg -------------------------------------------------------------------------------- /pipeline/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pipeline/attention_processors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/pipeline/attention_processors.py -------------------------------------------------------------------------------- /pipeline/attention_store_counting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/pipeline/attention_store_counting.py -------------------------------------------------------------------------------- /pipeline/countgen.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/pipeline/countgen.ipynb -------------------------------------------------------------------------------- /pipeline/mask_extraction/dbscan_mask_extract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/pipeline/mask_extraction/dbscan_mask_extract.py -------------------------------------------------------------------------------- /pipeline/mask_extraction/extract_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/pipeline/mask_extraction/extract_mask.py -------------------------------------------------------------------------------- /pipeline/mask_extraction/postprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/pipeline/mask_extraction/postprocess.py -------------------------------------------------------------------------------- /pipeline/mask_extraction/relayout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/pipeline/mask_extraction/relayout.py -------------------------------------------------------------------------------- /pipeline/mask_extraction/utils_masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/pipeline/mask_extraction/utils_masks.py -------------------------------------------------------------------------------- /pipeline/pipeline_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/pipeline/pipeline_config.yaml -------------------------------------------------------------------------------- /pipeline/run_countgen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/pipeline/run_countgen.py -------------------------------------------------------------------------------- /pipeline/self_counting_sdxl_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/pipeline/self_counting_sdxl_pipeline.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_relayout/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train_relayout/data_creation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train_relayout/data_creation/generate_counting_unet_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/train_relayout/data_creation/generate_counting_unet_data.py -------------------------------------------------------------------------------- /train_relayout/data_creation/matching_algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/train_relayout/data_creation/matching_algorithm.py -------------------------------------------------------------------------------- /train_relayout/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/train_relayout/dataloader.py -------------------------------------------------------------------------------- /train_relayout/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/train_relayout/loss.py -------------------------------------------------------------------------------- /train_relayout/relayout_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/train_relayout/relayout_requirements.txt -------------------------------------------------------------------------------- /train_relayout/train_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/train_relayout/train_unet.py -------------------------------------------------------------------------------- /train_relayout/unet_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/train_relayout/unet_config.yaml -------------------------------------------------------------------------------- /train_relayout/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/train_relayout/utils.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/attention_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/utils/attention_utils.py -------------------------------------------------------------------------------- /utils/counting_words_extract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/utils/counting_words_extract.py -------------------------------------------------------------------------------- /utils/generate_random_masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/utils/generate_random_masks.py -------------------------------------------------------------------------------- /utils/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Litalby1/make-it-count/HEAD/utils/loss_utils.py --------------------------------------------------------------------------------