├── .gitignore ├── LICENSE ├── README.md ├── T2IBenchmark ├── __init__.py ├── datasets │ ├── __init__.py │ └── coco2014 │ │ ├── __init__.py │ │ └── dataset.py ├── feature_extractors │ ├── __init__.py │ ├── base_feature_extractor.py │ ├── inceptionV3.py │ └── inceptionV3_feature_extractor.py ├── loaders.py ├── metrics │ ├── __init__.py │ └── fid.py ├── model_wrapper.py ├── models │ ├── __init__.py │ └── kandinsky21 │ │ ├── __init__.py │ │ ├── requirements.txt │ │ └── wrapper.py ├── pipelines.py └── utils │ ├── __init__.py │ ├── defaults.py │ ├── helpers.py │ ├── resizers.py │ └── seed_setter.py ├── assets ├── images │ ├── cats │ │ ├── 0.jpg │ │ ├── 1.jpg │ │ ├── 2.jpg │ │ ├── 3.jpg │ │ └── 4.jpg │ └── dogs │ │ ├── 0.jpg │ │ ├── 1.jpg │ │ ├── 2.jpg │ │ ├── 3.jpg │ │ └── 4.jpg └── logo.png ├── docs └── FID.md ├── examples ├── CLIP_score_usage.ipynb ├── FID-30k_on_MS-COCO.ipynb ├── FID_advanced.ipynb ├── FID_basic.ipynb └── ModelWrapper_FID-30k.ipynb ├── experiments └── resize_modes.ipynb ├── requirements.txt ├── setup.py └── tests ├── run_all_tests.py ├── test_frechet_distance.py └── test_resizer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/README.md -------------------------------------------------------------------------------- /T2IBenchmark/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/__init__.py -------------------------------------------------------------------------------- /T2IBenchmark/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/datasets/__init__.py -------------------------------------------------------------------------------- /T2IBenchmark/datasets/coco2014/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/datasets/coco2014/__init__.py -------------------------------------------------------------------------------- /T2IBenchmark/datasets/coco2014/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/datasets/coco2014/dataset.py -------------------------------------------------------------------------------- /T2IBenchmark/feature_extractors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/feature_extractors/__init__.py -------------------------------------------------------------------------------- /T2IBenchmark/feature_extractors/base_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/feature_extractors/base_feature_extractor.py -------------------------------------------------------------------------------- /T2IBenchmark/feature_extractors/inceptionV3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/feature_extractors/inceptionV3.py -------------------------------------------------------------------------------- /T2IBenchmark/feature_extractors/inceptionV3_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/feature_extractors/inceptionV3_feature_extractor.py -------------------------------------------------------------------------------- /T2IBenchmark/loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/loaders.py -------------------------------------------------------------------------------- /T2IBenchmark/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | from .fid import FIDStats, frechet_distance -------------------------------------------------------------------------------- /T2IBenchmark/metrics/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/metrics/fid.py -------------------------------------------------------------------------------- /T2IBenchmark/model_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/model_wrapper.py -------------------------------------------------------------------------------- /T2IBenchmark/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /T2IBenchmark/models/kandinsky21/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/models/kandinsky21/__init__.py -------------------------------------------------------------------------------- /T2IBenchmark/models/kandinsky21/requirements.txt: -------------------------------------------------------------------------------- 1 | git+https://github.com/ai-forever/Kandinsky-2.git -------------------------------------------------------------------------------- /T2IBenchmark/models/kandinsky21/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/models/kandinsky21/wrapper.py -------------------------------------------------------------------------------- /T2IBenchmark/pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/pipelines.py -------------------------------------------------------------------------------- /T2IBenchmark/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/utils/__init__.py -------------------------------------------------------------------------------- /T2IBenchmark/utils/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/utils/defaults.py -------------------------------------------------------------------------------- /T2IBenchmark/utils/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/utils/helpers.py -------------------------------------------------------------------------------- /T2IBenchmark/utils/resizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/utils/resizers.py -------------------------------------------------------------------------------- /T2IBenchmark/utils/seed_setter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/T2IBenchmark/utils/seed_setter.py -------------------------------------------------------------------------------- /assets/images/cats/0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/assets/images/cats/0.jpg -------------------------------------------------------------------------------- /assets/images/cats/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/assets/images/cats/1.jpg -------------------------------------------------------------------------------- /assets/images/cats/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/assets/images/cats/2.jpg -------------------------------------------------------------------------------- /assets/images/cats/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/assets/images/cats/3.jpg -------------------------------------------------------------------------------- /assets/images/cats/4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/assets/images/cats/4.jpg -------------------------------------------------------------------------------- /assets/images/dogs/0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/assets/images/dogs/0.jpg -------------------------------------------------------------------------------- /assets/images/dogs/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/assets/images/dogs/1.jpg -------------------------------------------------------------------------------- /assets/images/dogs/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/assets/images/dogs/2.jpg -------------------------------------------------------------------------------- /assets/images/dogs/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/assets/images/dogs/3.jpg -------------------------------------------------------------------------------- /assets/images/dogs/4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/assets/images/dogs/4.jpg -------------------------------------------------------------------------------- /assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/assets/logo.png -------------------------------------------------------------------------------- /docs/FID.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/docs/FID.md -------------------------------------------------------------------------------- /examples/CLIP_score_usage.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/examples/CLIP_score_usage.ipynb -------------------------------------------------------------------------------- /examples/FID-30k_on_MS-COCO.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/examples/FID-30k_on_MS-COCO.ipynb -------------------------------------------------------------------------------- /examples/FID_advanced.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/examples/FID_advanced.ipynb -------------------------------------------------------------------------------- /examples/FID_basic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/examples/FID_basic.ipynb -------------------------------------------------------------------------------- /examples/ModelWrapper_FID-30k.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/examples/ModelWrapper_FID-30k.ipynb -------------------------------------------------------------------------------- /experiments/resize_modes.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/experiments/resize_modes.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/setup.py -------------------------------------------------------------------------------- /tests/run_all_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/tests/run_all_tests.py -------------------------------------------------------------------------------- /tests/test_frechet_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/tests/test_frechet_distance.py -------------------------------------------------------------------------------- /tests/test_resizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/boomb0om/text2image-benchmark/HEAD/tests/test_resizer.py --------------------------------------------------------------------------------