├── .gitignore ├── LICENSE ├── README.md ├── configs └── spai.yaml ├── data ├── fake_dalle2.csv ├── fake_dalle3.csv ├── fake_firefly.csv ├── fake_flux.csv ├── fake_gigagan.csv ├── fake_glide.csv ├── fake_mjv5.csv ├── fake_mjv61.csv ├── fake_sd13.csv ├── fake_sd14.csv ├── fake_sd2.csv ├── fake_sd3.csv ├── fake_sdxl.csv ├── real_coco.csv ├── real_fodb.csv ├── real_imagenet.csv ├── real_openimages.csv └── real_raise.csv ├── docs ├── architecture.svg ├── data.md └── overview.svg ├── requirements.txt ├── spai ├── __init__.py ├── __main__.py ├── config.py ├── data │ ├── __init__.py │ ├── blur_kernels.py │ ├── data_finetune.py │ ├── data_mfm.py │ ├── filestorage.py │ ├── random_degradations.py │ └── readers.py ├── data_utils.py ├── logger.py ├── lr_scheduler.py ├── main_mfm.py ├── metrics.py ├── models │ ├── __init__.py │ ├── backbones.py │ ├── build.py │ ├── filters.py │ ├── frequency_loss.py │ ├── losses.py │ ├── mfm.py │ ├── sid.py │ ├── swin_transformer.py │ ├── utils.py │ └── vision_transformer.py ├── onnx.py ├── optimizer.py ├── tools │ ├── __init__.py │ ├── augment_dataset.py │ ├── create_dir_csv.py │ ├── create_dmid_ldm_train_val_csv.py │ ├── create_synthbuster_csv.py │ └── reduce_csv_column.py ├── tsne.py └── utils.py └── tests ├── __init__.py ├── data ├── __init__.py └── test_data_finetune.py └── models ├── __init__.py ├── test_backbones.py ├── test_filters.py ├── test_sid.py └── test_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/README.md -------------------------------------------------------------------------------- /configs/spai.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/configs/spai.yaml -------------------------------------------------------------------------------- /data/fake_dalle2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/fake_dalle2.csv -------------------------------------------------------------------------------- /data/fake_dalle3.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/fake_dalle3.csv -------------------------------------------------------------------------------- /data/fake_firefly.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/fake_firefly.csv -------------------------------------------------------------------------------- /data/fake_flux.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/fake_flux.csv -------------------------------------------------------------------------------- /data/fake_gigagan.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/fake_gigagan.csv -------------------------------------------------------------------------------- /data/fake_glide.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/fake_glide.csv -------------------------------------------------------------------------------- /data/fake_mjv5.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/fake_mjv5.csv -------------------------------------------------------------------------------- /data/fake_mjv61.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/fake_mjv61.csv -------------------------------------------------------------------------------- /data/fake_sd13.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/fake_sd13.csv -------------------------------------------------------------------------------- /data/fake_sd14.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/fake_sd14.csv -------------------------------------------------------------------------------- /data/fake_sd2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/fake_sd2.csv -------------------------------------------------------------------------------- /data/fake_sd3.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/fake_sd3.csv -------------------------------------------------------------------------------- /data/fake_sdxl.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/fake_sdxl.csv -------------------------------------------------------------------------------- /data/real_coco.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/real_coco.csv -------------------------------------------------------------------------------- /data/real_fodb.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/real_fodb.csv -------------------------------------------------------------------------------- /data/real_imagenet.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/real_imagenet.csv -------------------------------------------------------------------------------- /data/real_openimages.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/real_openimages.csv -------------------------------------------------------------------------------- /data/real_raise.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/data/real_raise.csv -------------------------------------------------------------------------------- /docs/architecture.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/docs/architecture.svg -------------------------------------------------------------------------------- /docs/data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/docs/data.md -------------------------------------------------------------------------------- /docs/overview.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/docs/overview.svg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/requirements.txt -------------------------------------------------------------------------------- /spai/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/__init__.py -------------------------------------------------------------------------------- /spai/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/__main__.py -------------------------------------------------------------------------------- /spai/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/config.py -------------------------------------------------------------------------------- /spai/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/data/__init__.py -------------------------------------------------------------------------------- /spai/data/blur_kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/data/blur_kernels.py -------------------------------------------------------------------------------- /spai/data/data_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/data/data_finetune.py -------------------------------------------------------------------------------- /spai/data/data_mfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/data/data_mfm.py -------------------------------------------------------------------------------- /spai/data/filestorage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/data/filestorage.py -------------------------------------------------------------------------------- /spai/data/random_degradations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/data/random_degradations.py -------------------------------------------------------------------------------- /spai/data/readers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/data/readers.py -------------------------------------------------------------------------------- /spai/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/data_utils.py -------------------------------------------------------------------------------- /spai/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/logger.py -------------------------------------------------------------------------------- /spai/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/lr_scheduler.py -------------------------------------------------------------------------------- /spai/main_mfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/main_mfm.py -------------------------------------------------------------------------------- /spai/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/metrics.py -------------------------------------------------------------------------------- /spai/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/models/__init__.py -------------------------------------------------------------------------------- /spai/models/backbones.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/models/backbones.py -------------------------------------------------------------------------------- /spai/models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/models/build.py -------------------------------------------------------------------------------- /spai/models/filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/models/filters.py -------------------------------------------------------------------------------- /spai/models/frequency_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/models/frequency_loss.py -------------------------------------------------------------------------------- /spai/models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/models/losses.py -------------------------------------------------------------------------------- /spai/models/mfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/models/mfm.py -------------------------------------------------------------------------------- /spai/models/sid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/models/sid.py -------------------------------------------------------------------------------- /spai/models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/models/swin_transformer.py -------------------------------------------------------------------------------- /spai/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/models/utils.py -------------------------------------------------------------------------------- /spai/models/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/models/vision_transformer.py -------------------------------------------------------------------------------- /spai/onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/onnx.py -------------------------------------------------------------------------------- /spai/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/optimizer.py -------------------------------------------------------------------------------- /spai/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/tools/__init__.py -------------------------------------------------------------------------------- /spai/tools/augment_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/tools/augment_dataset.py -------------------------------------------------------------------------------- /spai/tools/create_dir_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/tools/create_dir_csv.py -------------------------------------------------------------------------------- /spai/tools/create_dmid_ldm_train_val_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/tools/create_dmid_ldm_train_val_csv.py -------------------------------------------------------------------------------- /spai/tools/create_synthbuster_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/tools/create_synthbuster_csv.py -------------------------------------------------------------------------------- /spai/tools/reduce_csv_column.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/tools/reduce_csv_column.py -------------------------------------------------------------------------------- /spai/tsne.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/tsne.py -------------------------------------------------------------------------------- /spai/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/spai/utils.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/tests/data/__init__.py -------------------------------------------------------------------------------- /tests/data/test_data_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/tests/data/test_data_finetune.py -------------------------------------------------------------------------------- /tests/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/tests/models/__init__.py -------------------------------------------------------------------------------- /tests/models/test_backbones.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/tests/models/test_backbones.py -------------------------------------------------------------------------------- /tests/models/test_filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/tests/models/test_filters.py -------------------------------------------------------------------------------- /tests/models/test_sid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/tests/models/test_sid.py -------------------------------------------------------------------------------- /tests/models/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mever-team/spai/HEAD/tests/models/test_utils.py --------------------------------------------------------------------------------