├── LICENSE ├── README.md ├── common.py ├── data ├── __init__.py ├── aligned_dataset.py ├── base_dataset.py ├── cityscapes_dataset.py ├── image_folder.py ├── single_dataset.py ├── spade_dataset.py ├── template_dataset.py └── unaligned_dataset.py ├── datasets ├── combine_A_and_B.py ├── download_cyclegan_dataset.sh ├── download_pix2pix_dataset.sh ├── get_trainIds.py ├── make_dataset_aligned.py ├── prepare_cityscapes_dataset.py └── table.txt ├── distill.py ├── distillers ├── __init__.py ├── base_inception_distiller.py ├── base_spade_distiller.py ├── inception_distiller.py └── spade_distiller.py ├── environment.yaml ├── get_real_stat.py ├── images ├── anime.gif ├── comparison.png ├── input.gif ├── night.gif └── style.gif ├── metric ├── __init__.py ├── cityscapes_mIoU.py ├── deeplabv2.py ├── drn.py ├── fid_score.py ├── inception.py ├── kid_score.py └── mIoU_score.py ├── models ├── __init__.py ├── base_model.py ├── cycle_gan_model.py ├── modules │ ├── __init__.py │ ├── discriminators.py │ ├── inception_architecture │ │ ├── __init__.py │ │ ├── inception_generator.py │ │ └── inception_spade_generator.py │ ├── inception_modules.py │ ├── loss.py │ ├── spade_architecture │ │ ├── __init__.py │ │ └── normalization.py │ ├── spade_modules │ │ ├── __init__.py │ │ ├── base_spade_distiller_modules.py │ │ ├── spade_distiller_modules.py │ │ └── spade_model_modules.py │ └── sync_batchnorm │ │ ├── __init__.py │ │ ├── batchnorm.py │ │ ├── batchnorm_reimpl.py │ │ ├── comm.py │ │ ├── replicate.py │ │ └── unittest.py ├── networks.py ├── pix2pix_model.py └── spade_model.py ├── onnx_export.py ├── onnx_exporter.py ├── options ├── __init__.py ├── base_options.py ├── distill_options.py ├── test_options.py └── train_options.py ├── profile.py ├── profiler.py ├── requirements.in ├── requirements.txt ├── scripts ├── cycle_gan │ ├── horse2zebra │ │ ├── evaluate_inception_student_2p6B.sh │ │ ├── onnx_export_inception_student_2p6B.sh │ │ ├── train_inception_student_2p6B.sh │ │ └── train_inception_teacher.sh │ └── zebra2horse │ │ ├── evaluate_inception_student_2p6B.sh │ │ ├── onnx_export_inception_student_2p6B.sh │ │ ├── train_inception_student_2p6B.sh │ │ └── train_inception_teacher.sh ├── gaugan │ └── cityscapes │ │ ├── evaluate_inception_student_30B.sh │ │ ├── evaluate_inception_student_5p6B.sh │ │ ├── onnx_export_inception_student_30B.sh │ │ ├── onnx_export_inception_student_5p6B.sh │ │ ├── train_inception_student_30B.sh │ │ ├── train_inception_student_5p6B.sh │ │ └── train_inception_teacher.sh └── pix2pix │ ├── cityscapes │ ├── evaluate_inception_student_5p6B.sh │ ├── onnx_export_inception_student_5p6B.sh │ ├── train_inception_student_5p6B.sh │ └── train_inception_teacher.sh │ └── map2sat │ ├── evaluate_inception_student_4p6B.sh │ ├── onnx_export_inception_student_4p6B.sh │ ├── train_inception_student_4p6B.sh │ └── train_inception_teacher.sh ├── train.py ├── trainer.py └── utils ├── __init__.py ├── common.py ├── html.py ├── image_pool.py ├── logger.py ├── model_profiling.py ├── prune.py ├── util.py └── weight_transfer.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/README.md -------------------------------------------------------------------------------- /common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/common.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/aligned_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/data/aligned_dataset.py -------------------------------------------------------------------------------- /data/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/data/base_dataset.py -------------------------------------------------------------------------------- /data/cityscapes_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/data/cityscapes_dataset.py -------------------------------------------------------------------------------- /data/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/data/image_folder.py -------------------------------------------------------------------------------- /data/single_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/data/single_dataset.py -------------------------------------------------------------------------------- /data/spade_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/data/spade_dataset.py -------------------------------------------------------------------------------- /data/template_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/data/template_dataset.py -------------------------------------------------------------------------------- /data/unaligned_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/data/unaligned_dataset.py -------------------------------------------------------------------------------- /datasets/combine_A_and_B.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/datasets/combine_A_and_B.py -------------------------------------------------------------------------------- /datasets/download_cyclegan_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/datasets/download_cyclegan_dataset.sh -------------------------------------------------------------------------------- /datasets/download_pix2pix_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/datasets/download_pix2pix_dataset.sh -------------------------------------------------------------------------------- /datasets/get_trainIds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/datasets/get_trainIds.py -------------------------------------------------------------------------------- /datasets/make_dataset_aligned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/datasets/make_dataset_aligned.py -------------------------------------------------------------------------------- /datasets/prepare_cityscapes_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/datasets/prepare_cityscapes_dataset.py -------------------------------------------------------------------------------- /datasets/table.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/datasets/table.txt -------------------------------------------------------------------------------- /distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/distill.py -------------------------------------------------------------------------------- /distillers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/distillers/__init__.py -------------------------------------------------------------------------------- /distillers/base_inception_distiller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/distillers/base_inception_distiller.py -------------------------------------------------------------------------------- /distillers/base_spade_distiller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/distillers/base_spade_distiller.py -------------------------------------------------------------------------------- /distillers/inception_distiller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/distillers/inception_distiller.py -------------------------------------------------------------------------------- /distillers/spade_distiller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/distillers/spade_distiller.py -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/environment.yaml -------------------------------------------------------------------------------- /get_real_stat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/get_real_stat.py -------------------------------------------------------------------------------- /images/anime.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/images/anime.gif -------------------------------------------------------------------------------- /images/comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/images/comparison.png -------------------------------------------------------------------------------- /images/input.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/images/input.gif -------------------------------------------------------------------------------- /images/night.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/images/night.gif -------------------------------------------------------------------------------- /images/style.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/images/style.gif -------------------------------------------------------------------------------- /metric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/metric/__init__.py -------------------------------------------------------------------------------- /metric/cityscapes_mIoU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/metric/cityscapes_mIoU.py -------------------------------------------------------------------------------- /metric/deeplabv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/metric/deeplabv2.py -------------------------------------------------------------------------------- /metric/drn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/metric/drn.py -------------------------------------------------------------------------------- /metric/fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/metric/fid_score.py -------------------------------------------------------------------------------- /metric/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/metric/inception.py -------------------------------------------------------------------------------- /metric/kid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/metric/kid_score.py -------------------------------------------------------------------------------- /metric/mIoU_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/metric/mIoU_score.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/base_model.py -------------------------------------------------------------------------------- /models/cycle_gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/cycle_gan_model.py -------------------------------------------------------------------------------- /models/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/modules/discriminators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/discriminators.py -------------------------------------------------------------------------------- /models/modules/inception_architecture/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/modules/inception_architecture/inception_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/inception_architecture/inception_generator.py -------------------------------------------------------------------------------- /models/modules/inception_architecture/inception_spade_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/inception_architecture/inception_spade_generator.py -------------------------------------------------------------------------------- /models/modules/inception_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/inception_modules.py -------------------------------------------------------------------------------- /models/modules/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/loss.py -------------------------------------------------------------------------------- /models/modules/spade_architecture/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/modules/spade_architecture/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/spade_architecture/normalization.py -------------------------------------------------------------------------------- /models/modules/spade_modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/modules/spade_modules/base_spade_distiller_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/spade_modules/base_spade_distiller_modules.py -------------------------------------------------------------------------------- /models/modules/spade_modules/spade_distiller_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/spade_modules/spade_distiller_modules.py -------------------------------------------------------------------------------- /models/modules/spade_modules/spade_model_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/spade_modules/spade_model_modules.py -------------------------------------------------------------------------------- /models/modules/sync_batchnorm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/sync_batchnorm/__init__.py -------------------------------------------------------------------------------- /models/modules/sync_batchnorm/batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/sync_batchnorm/batchnorm.py -------------------------------------------------------------------------------- /models/modules/sync_batchnorm/batchnorm_reimpl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/sync_batchnorm/batchnorm_reimpl.py -------------------------------------------------------------------------------- /models/modules/sync_batchnorm/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/sync_batchnorm/comm.py -------------------------------------------------------------------------------- /models/modules/sync_batchnorm/replicate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/sync_batchnorm/replicate.py -------------------------------------------------------------------------------- /models/modules/sync_batchnorm/unittest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/modules/sync_batchnorm/unittest.py -------------------------------------------------------------------------------- /models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/networks.py -------------------------------------------------------------------------------- /models/pix2pix_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/pix2pix_model.py -------------------------------------------------------------------------------- /models/spade_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/models/spade_model.py -------------------------------------------------------------------------------- /onnx_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/onnx_export.py -------------------------------------------------------------------------------- /onnx_exporter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/onnx_exporter.py -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/options/__init__.py -------------------------------------------------------------------------------- /options/base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/options/base_options.py -------------------------------------------------------------------------------- /options/distill_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/options/distill_options.py -------------------------------------------------------------------------------- /options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/options/test_options.py -------------------------------------------------------------------------------- /options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/options/train_options.py -------------------------------------------------------------------------------- /profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/profile.py -------------------------------------------------------------------------------- /profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/profiler.py -------------------------------------------------------------------------------- /requirements.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/requirements.in -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/cycle_gan/horse2zebra/evaluate_inception_student_2p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/cycle_gan/horse2zebra/evaluate_inception_student_2p6B.sh -------------------------------------------------------------------------------- /scripts/cycle_gan/horse2zebra/onnx_export_inception_student_2p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/cycle_gan/horse2zebra/onnx_export_inception_student_2p6B.sh -------------------------------------------------------------------------------- /scripts/cycle_gan/horse2zebra/train_inception_student_2p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/cycle_gan/horse2zebra/train_inception_student_2p6B.sh -------------------------------------------------------------------------------- /scripts/cycle_gan/horse2zebra/train_inception_teacher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/cycle_gan/horse2zebra/train_inception_teacher.sh -------------------------------------------------------------------------------- /scripts/cycle_gan/zebra2horse/evaluate_inception_student_2p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/cycle_gan/zebra2horse/evaluate_inception_student_2p6B.sh -------------------------------------------------------------------------------- /scripts/cycle_gan/zebra2horse/onnx_export_inception_student_2p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/cycle_gan/zebra2horse/onnx_export_inception_student_2p6B.sh -------------------------------------------------------------------------------- /scripts/cycle_gan/zebra2horse/train_inception_student_2p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/cycle_gan/zebra2horse/train_inception_student_2p6B.sh -------------------------------------------------------------------------------- /scripts/cycle_gan/zebra2horse/train_inception_teacher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/cycle_gan/zebra2horse/train_inception_teacher.sh -------------------------------------------------------------------------------- /scripts/gaugan/cityscapes/evaluate_inception_student_30B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/gaugan/cityscapes/evaluate_inception_student_30B.sh -------------------------------------------------------------------------------- /scripts/gaugan/cityscapes/evaluate_inception_student_5p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/gaugan/cityscapes/evaluate_inception_student_5p6B.sh -------------------------------------------------------------------------------- /scripts/gaugan/cityscapes/onnx_export_inception_student_30B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/gaugan/cityscapes/onnx_export_inception_student_30B.sh -------------------------------------------------------------------------------- /scripts/gaugan/cityscapes/onnx_export_inception_student_5p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/gaugan/cityscapes/onnx_export_inception_student_5p6B.sh -------------------------------------------------------------------------------- /scripts/gaugan/cityscapes/train_inception_student_30B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/gaugan/cityscapes/train_inception_student_30B.sh -------------------------------------------------------------------------------- /scripts/gaugan/cityscapes/train_inception_student_5p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/gaugan/cityscapes/train_inception_student_5p6B.sh -------------------------------------------------------------------------------- /scripts/gaugan/cityscapes/train_inception_teacher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/gaugan/cityscapes/train_inception_teacher.sh -------------------------------------------------------------------------------- /scripts/pix2pix/cityscapes/evaluate_inception_student_5p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/pix2pix/cityscapes/evaluate_inception_student_5p6B.sh -------------------------------------------------------------------------------- /scripts/pix2pix/cityscapes/onnx_export_inception_student_5p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/pix2pix/cityscapes/onnx_export_inception_student_5p6B.sh -------------------------------------------------------------------------------- /scripts/pix2pix/cityscapes/train_inception_student_5p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/pix2pix/cityscapes/train_inception_student_5p6B.sh -------------------------------------------------------------------------------- /scripts/pix2pix/cityscapes/train_inception_teacher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/pix2pix/cityscapes/train_inception_teacher.sh -------------------------------------------------------------------------------- /scripts/pix2pix/map2sat/evaluate_inception_student_4p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/pix2pix/map2sat/evaluate_inception_student_4p6B.sh -------------------------------------------------------------------------------- /scripts/pix2pix/map2sat/onnx_export_inception_student_4p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/pix2pix/map2sat/onnx_export_inception_student_4p6B.sh -------------------------------------------------------------------------------- /scripts/pix2pix/map2sat/train_inception_student_4p6B.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/pix2pix/map2sat/train_inception_student_4p6B.sh -------------------------------------------------------------------------------- /scripts/pix2pix/map2sat/train_inception_teacher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/scripts/pix2pix/map2sat/train_inception_teacher.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/train.py -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/trainer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/utils/common.py -------------------------------------------------------------------------------- /utils/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/utils/html.py -------------------------------------------------------------------------------- /utils/image_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/utils/image_pool.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/model_profiling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/utils/model_profiling.py -------------------------------------------------------------------------------- /utils/prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/utils/prune.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/utils/util.py -------------------------------------------------------------------------------- /utils/weight_transfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-research/CAT/HEAD/utils/weight_transfer.py --------------------------------------------------------------------------------