├── .gitignore ├── LICENSE ├── README.md ├── dataset_tool.py ├── datasets ├── dataset_512.py ├── dataset_512_val.py ├── mask_generator_512.py └── mask_generator_512_small.py ├── dnnlib ├── __init__.py └── util.py ├── evaluatoin ├── cal_fid_pids_uids.py ├── cal_lpips.py └── cal_psnr_ssim_l1.py ├── generate_image.py ├── legacy.py ├── losses ├── loss.py ├── pcp.py └── vggNet.py ├── metrics ├── __init__.py ├── frechet_inception_distance.py ├── inception_discriminative_score.py ├── inception_score.py ├── kernel_inception_distance.py ├── metric_main.py ├── metric_utils.py ├── perceptual_path_length.py ├── precision_recall.py └── psnr_ssim_l1.py ├── networks ├── basic_module.py └── mat.py ├── requirements.txt ├── test_sets ├── CelebA-HQ │ ├── images │ │ ├── test1.png │ │ └── test2.png │ └── masks │ │ ├── mask1.png │ │ └── mask2.png └── Places │ ├── images │ ├── test1.jpg │ └── test2.jpg │ └── masks │ ├── mask1.png │ └── mask2.png ├── torch_utils ├── __init__.py ├── custom_ops.py ├── misc.py ├── ops │ ├── __init__.py │ ├── bias_act.cpp │ ├── bias_act.cu │ ├── bias_act.h │ ├── bias_act.py │ ├── conv2d_gradfix.py │ ├── conv2d_resample.py │ ├── fma.py │ ├── grid_sample_gradfix.py │ ├── upfirdn2d.cpp │ ├── upfirdn2d.cu │ ├── upfirdn2d.h │ └── upfirdn2d.py ├── persistence.py └── training_stats.py ├── train.py └── training ├── __init__.py ├── augment.py └── training_loop.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/README.md -------------------------------------------------------------------------------- /dataset_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/dataset_tool.py -------------------------------------------------------------------------------- /datasets/dataset_512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/datasets/dataset_512.py -------------------------------------------------------------------------------- /datasets/dataset_512_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/datasets/dataset_512_val.py -------------------------------------------------------------------------------- /datasets/mask_generator_512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/datasets/mask_generator_512.py -------------------------------------------------------------------------------- /datasets/mask_generator_512_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/datasets/mask_generator_512_small.py -------------------------------------------------------------------------------- /dnnlib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/dnnlib/__init__.py -------------------------------------------------------------------------------- /dnnlib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/dnnlib/util.py -------------------------------------------------------------------------------- /evaluatoin/cal_fid_pids_uids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/evaluatoin/cal_fid_pids_uids.py -------------------------------------------------------------------------------- /evaluatoin/cal_lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/evaluatoin/cal_lpips.py -------------------------------------------------------------------------------- /evaluatoin/cal_psnr_ssim_l1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/evaluatoin/cal_psnr_ssim_l1.py -------------------------------------------------------------------------------- /generate_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/generate_image.py -------------------------------------------------------------------------------- /legacy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/legacy.py -------------------------------------------------------------------------------- /losses/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/losses/loss.py -------------------------------------------------------------------------------- /losses/pcp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/losses/pcp.py -------------------------------------------------------------------------------- /losses/vggNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/losses/vggNet.py -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/metrics/__init__.py -------------------------------------------------------------------------------- /metrics/frechet_inception_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/metrics/frechet_inception_distance.py -------------------------------------------------------------------------------- /metrics/inception_discriminative_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/metrics/inception_discriminative_score.py -------------------------------------------------------------------------------- /metrics/inception_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/metrics/inception_score.py -------------------------------------------------------------------------------- /metrics/kernel_inception_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/metrics/kernel_inception_distance.py -------------------------------------------------------------------------------- /metrics/metric_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/metrics/metric_main.py -------------------------------------------------------------------------------- /metrics/metric_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/metrics/metric_utils.py -------------------------------------------------------------------------------- /metrics/perceptual_path_length.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/metrics/perceptual_path_length.py -------------------------------------------------------------------------------- /metrics/precision_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/metrics/precision_recall.py -------------------------------------------------------------------------------- /metrics/psnr_ssim_l1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/metrics/psnr_ssim_l1.py -------------------------------------------------------------------------------- /networks/basic_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/networks/basic_module.py -------------------------------------------------------------------------------- /networks/mat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/networks/mat.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/requirements.txt -------------------------------------------------------------------------------- /test_sets/CelebA-HQ/images/test1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/test_sets/CelebA-HQ/images/test1.png -------------------------------------------------------------------------------- /test_sets/CelebA-HQ/images/test2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/test_sets/CelebA-HQ/images/test2.png -------------------------------------------------------------------------------- /test_sets/CelebA-HQ/masks/mask1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/test_sets/CelebA-HQ/masks/mask1.png -------------------------------------------------------------------------------- /test_sets/CelebA-HQ/masks/mask2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/test_sets/CelebA-HQ/masks/mask2.png -------------------------------------------------------------------------------- /test_sets/Places/images/test1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/test_sets/Places/images/test1.jpg -------------------------------------------------------------------------------- /test_sets/Places/images/test2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/test_sets/Places/images/test2.jpg -------------------------------------------------------------------------------- /test_sets/Places/masks/mask1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/test_sets/Places/masks/mask1.png -------------------------------------------------------------------------------- /test_sets/Places/masks/mask2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/test_sets/Places/masks/mask2.png -------------------------------------------------------------------------------- /torch_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/__init__.py -------------------------------------------------------------------------------- /torch_utils/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/custom_ops.py -------------------------------------------------------------------------------- /torch_utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/misc.py -------------------------------------------------------------------------------- /torch_utils/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/ops/__init__.py -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/ops/bias_act.cpp -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/ops/bias_act.cu -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/ops/bias_act.h -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/ops/bias_act.py -------------------------------------------------------------------------------- /torch_utils/ops/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/ops/conv2d_gradfix.py -------------------------------------------------------------------------------- /torch_utils/ops/conv2d_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/ops/conv2d_resample.py -------------------------------------------------------------------------------- /torch_utils/ops/fma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/ops/fma.py -------------------------------------------------------------------------------- /torch_utils/ops/grid_sample_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/ops/grid_sample_gradfix.py -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/ops/upfirdn2d.cpp -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/ops/upfirdn2d.cu -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/ops/upfirdn2d.h -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/ops/upfirdn2d.py -------------------------------------------------------------------------------- /torch_utils/persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/persistence.py -------------------------------------------------------------------------------- /torch_utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/torch_utils/training_stats.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/train.py -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/training/__init__.py -------------------------------------------------------------------------------- /training/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/training/augment.py -------------------------------------------------------------------------------- /training/training_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/MAT/HEAD/training/training_loop.py --------------------------------------------------------------------------------