├── .gitignore ├── LICENSE ├── README.md ├── colab └── FcF_Inpainting.ipynb ├── colab_demo.py ├── datasets └── demo │ ├── celeba-hq │ ├── facea.png │ ├── facea_mask.png │ ├── faceb.png │ ├── faceb_mask.png │ ├── facec.png │ ├── facec_mask.png │ ├── faced.png │ ├── faced_mask.png │ ├── facee.png │ ├── facee_mask.png │ ├── facef.png │ ├── facef_mask.png │ ├── faceg.png │ ├── faceg_mask.png │ ├── faceh.png │ ├── faceh_mask.png │ ├── facei.png │ ├── facei_mask.png │ ├── facej.png │ ├── facej_mask.png │ ├── facek.png │ ├── facek_mask.png │ ├── facel.png │ └── facel_mask.png │ └── places2 │ ├── 082_Places365_val_00021838.png │ ├── 082_Places365_val_00021838_mask.png │ ├── 090_Places365_val_00006548.png │ ├── 090_Places365_val_00006548_mask.png │ ├── 120_Places365_val_00000241.png │ ├── 120_Places365_val_00000241_mask.png │ ├── 129_Places365_val_00012805.png │ ├── 129_Places365_val_00012805_mask.png │ ├── 148_Places365_val_00000059.png │ ├── 148_Places365_val_00000059_mask.png │ ├── 182_Places365_val_00016909.png │ ├── 182_Places365_val_00016909_mask.png │ ├── 200_Places365_val_00011651.png │ ├── 200_Places365_val_00011651_mask.png │ ├── 212_Places365_val_00014101.png │ └── 212_Places365_val_00014101_mask.png ├── demo.py ├── dnnlib ├── __init__.py └── util.py ├── docs ├── celeba_qual.svg ├── fcf_gan.svg └── places_qual.svg ├── evaluate.py ├── legacy.py ├── metrics ├── configs │ └── eval2_gpu.yaml ├── evaluate_predicts.py └── evaluation │ ├── __init__.py │ ├── data.py │ ├── evaluator.py │ ├── losses │ ├── __init__.py │ ├── base_loss.py │ ├── fid │ │ ├── __init__.py │ │ ├── fid_score.py │ │ └── inception.py │ ├── lpips.py │ └── ssim.py │ ├── masks │ ├── README.md │ ├── __init__.py │ ├── countless │ │ ├── .gitignore │ │ ├── README.md │ │ ├── __init__.py │ │ ├── countless2d.py │ │ ├── countless3d.py │ │ ├── images │ │ │ ├── gcim.jpg │ │ │ ├── gray_segmentation.png │ │ │ ├── segmentation.png │ │ │ └── sparse.png │ │ ├── memprof │ │ │ ├── countless2d_gcim_N_1000.png │ │ │ ├── countless2d_quick_gcim_N_1000.png │ │ │ ├── countless3d.png │ │ │ ├── countless3d_dynamic.png │ │ │ ├── countless3d_dynamic_generalized.png │ │ │ └── countless3d_generalized.png │ │ └── requirements.txt │ └── mask.py │ ├── utils.py │ └── vis.py ├── models └── lpips_models │ ├── alex.pth │ ├── squeeze.pth │ └── vgg.pth ├── requirements.txt ├── streamlit_demo ├── README.md ├── app.py ├── demo.png ├── dnnlib │ ├── __init__.py │ └── util.py ├── fcf_gan.png ├── legacy.py ├── packages.txt ├── requirements.txt ├── setup.sh ├── test_512 │ ├── church.jpeg │ ├── door.jpeg │ ├── floor.jpeg │ ├── house.jpeg │ ├── person-cliff.jpeg │ ├── person-fence.png │ ├── persons-white-fence.jpeg │ ├── wall-1.jpeg │ └── wall-2.jpeg ├── 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 └── training │ ├── __init__.py │ ├── data │ ├── augment.py │ ├── configs │ │ ├── medium_256.yaml │ │ ├── medium_512.yaml │ │ ├── segm_256.yaml │ │ ├── segm_512.yaml │ │ ├── thick_256.yaml │ │ ├── thick_512.yaml │ │ ├── thin_256.yaml │ │ └── thin_512.yaml │ ├── dataset.py │ ├── demo_loader.py │ ├── gen_loader.py │ ├── lama_mask_generator_test.py │ ├── mask_generator.py │ └── pred_loader.py │ ├── ffc.py │ ├── losses │ ├── ade20k │ │ ├── __init__.py │ │ ├── base.py │ │ ├── color150.mat │ │ ├── mobilenet.py │ │ ├── object150_info.csv │ │ ├── resnet.py │ │ ├── segm_lib.zip │ │ ├── segm_lib │ │ │ ├── nn │ │ │ │ ├── __init__.py │ │ │ │ ├── modules │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── batchnorm.py │ │ │ │ │ ├── comm.py │ │ │ │ │ ├── replicate.py │ │ │ │ │ ├── tests │ │ │ │ │ │ ├── test_numeric_batchnorm.py │ │ │ │ │ │ └── test_sync_batchnorm.py │ │ │ │ │ └── unittest.py │ │ │ │ └── parallel │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── data_parallel.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── data │ │ │ │ ├── __init__.py │ │ │ │ ├── dataloader.py │ │ │ │ ├── dataset.py │ │ │ │ ├── distributed.py │ │ │ │ └── sampler.py │ │ │ │ └── th.py │ │ └── utils.py │ ├── high_receptive_pl.py │ ├── loss.py │ ├── perceptual.py │ └── vgg.py │ ├── models.py │ ├── networks.py │ └── training_loop.py ├── tools ├── eval_sampler.py ├── eval_segm_sampler.py ├── gen_masks.py ├── gen_masks.sh ├── gen_random_segm_masks.py ├── prepare_celebahq.sh ├── prepare_celebahq_evaluation.sh ├── prepare_places_evaluation.sh ├── prepare_places_segm_evaluation.sh ├── prepare_places_val.sh ├── train_shuffled.flist ├── val_sampler.py └── val_shuffled.flist ├── 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 ├── data │ ├── augment.py │ ├── configs │ │ ├── medium_256.yaml │ │ ├── medium_512.yaml │ │ ├── segm_256.yaml │ │ ├── segm_512.yaml │ │ ├── thick_256.yaml │ │ ├── thick_512.yaml │ │ ├── thin_256.yaml │ │ └── thin_512.yaml │ ├── dataset.py │ ├── demo_loader.py │ ├── gen_loader.py │ ├── lama_mask_generator_test.py │ ├── mask_generator.py │ └── pred_loader.py ├── ffc.py ├── losses │ ├── ade20k │ │ ├── __init__.py │ │ ├── base.py │ │ ├── color150.mat │ │ ├── mobilenet.py │ │ ├── object150_info.csv │ │ ├── resnet.py │ │ ├── segm_lib.zip │ │ ├── segm_lib │ │ │ ├── nn │ │ │ │ ├── __init__.py │ │ │ │ ├── modules │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── batchnorm.py │ │ │ │ │ ├── comm.py │ │ │ │ │ ├── replicate.py │ │ │ │ │ ├── tests │ │ │ │ │ │ ├── test_numeric_batchnorm.py │ │ │ │ │ │ └── test_sync_batchnorm.py │ │ │ │ │ └── unittest.py │ │ │ │ └── parallel │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── data_parallel.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── data │ │ │ │ ├── __init__.py │ │ │ │ ├── dataloader.py │ │ │ │ ├── dataset.py │ │ │ │ ├── distributed.py │ │ │ │ └── sampler.py │ │ │ │ └── th.py │ │ └── utils.py │ ├── high_receptive_pl.py │ ├── loss.py │ ├── perceptual.py │ └── vgg.py ├── models.py ├── networks.py └── training_loop.py └── utils ├── __init__.py └── data_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/README.md -------------------------------------------------------------------------------- /colab/FcF_Inpainting.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/colab/FcF_Inpainting.ipynb -------------------------------------------------------------------------------- /colab_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/colab_demo.py -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facea.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facea.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facea_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facea_mask.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/faceb.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/faceb.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/faceb_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/faceb_mask.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facec.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facec.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facec_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facec_mask.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/faced.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/faced.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/faced_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/faced_mask.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facee.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facee.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facee_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facee_mask.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facef.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facef.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facef_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facef_mask.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/faceg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/faceg.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/faceg_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/faceg_mask.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/faceh.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/faceh.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/faceh_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/faceh_mask.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facei.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facei.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facei_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facei_mask.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facej.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facej.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facej_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facej_mask.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facek.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facek.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facek_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facek_mask.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facel.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facel.png -------------------------------------------------------------------------------- /datasets/demo/celeba-hq/facel_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/celeba-hq/facel_mask.png -------------------------------------------------------------------------------- /datasets/demo/places2/082_Places365_val_00021838.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/082_Places365_val_00021838.png -------------------------------------------------------------------------------- /datasets/demo/places2/082_Places365_val_00021838_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/082_Places365_val_00021838_mask.png -------------------------------------------------------------------------------- /datasets/demo/places2/090_Places365_val_00006548.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/090_Places365_val_00006548.png -------------------------------------------------------------------------------- /datasets/demo/places2/090_Places365_val_00006548_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/090_Places365_val_00006548_mask.png -------------------------------------------------------------------------------- /datasets/demo/places2/120_Places365_val_00000241.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/120_Places365_val_00000241.png -------------------------------------------------------------------------------- /datasets/demo/places2/120_Places365_val_00000241_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/120_Places365_val_00000241_mask.png -------------------------------------------------------------------------------- /datasets/demo/places2/129_Places365_val_00012805.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/129_Places365_val_00012805.png -------------------------------------------------------------------------------- /datasets/demo/places2/129_Places365_val_00012805_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/129_Places365_val_00012805_mask.png -------------------------------------------------------------------------------- /datasets/demo/places2/148_Places365_val_00000059.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/148_Places365_val_00000059.png -------------------------------------------------------------------------------- /datasets/demo/places2/148_Places365_val_00000059_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/148_Places365_val_00000059_mask.png -------------------------------------------------------------------------------- /datasets/demo/places2/182_Places365_val_00016909.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/182_Places365_val_00016909.png -------------------------------------------------------------------------------- /datasets/demo/places2/182_Places365_val_00016909_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/182_Places365_val_00016909_mask.png -------------------------------------------------------------------------------- /datasets/demo/places2/200_Places365_val_00011651.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/200_Places365_val_00011651.png -------------------------------------------------------------------------------- /datasets/demo/places2/200_Places365_val_00011651_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/200_Places365_val_00011651_mask.png -------------------------------------------------------------------------------- /datasets/demo/places2/212_Places365_val_00014101.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/212_Places365_val_00014101.png -------------------------------------------------------------------------------- /datasets/demo/places2/212_Places365_val_00014101_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/datasets/demo/places2/212_Places365_val_00014101_mask.png -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/demo.py -------------------------------------------------------------------------------- /dnnlib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/dnnlib/__init__.py -------------------------------------------------------------------------------- /dnnlib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/dnnlib/util.py -------------------------------------------------------------------------------- /docs/celeba_qual.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/docs/celeba_qual.svg -------------------------------------------------------------------------------- /docs/fcf_gan.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/docs/fcf_gan.svg -------------------------------------------------------------------------------- /docs/places_qual.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/docs/places_qual.svg -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/evaluate.py -------------------------------------------------------------------------------- /legacy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/legacy.py -------------------------------------------------------------------------------- /metrics/configs/eval2_gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/configs/eval2_gpu.yaml -------------------------------------------------------------------------------- /metrics/evaluate_predicts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluate_predicts.py -------------------------------------------------------------------------------- /metrics/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/__init__.py -------------------------------------------------------------------------------- /metrics/evaluation/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/data.py -------------------------------------------------------------------------------- /metrics/evaluation/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/evaluator.py -------------------------------------------------------------------------------- /metrics/evaluation/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /metrics/evaluation/losses/base_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/losses/base_loss.py -------------------------------------------------------------------------------- /metrics/evaluation/losses/fid/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /metrics/evaluation/losses/fid/fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/losses/fid/fid_score.py -------------------------------------------------------------------------------- /metrics/evaluation/losses/fid/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/losses/fid/inception.py -------------------------------------------------------------------------------- /metrics/evaluation/losses/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/losses/lpips.py -------------------------------------------------------------------------------- /metrics/evaluation/losses/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/losses/ssim.py -------------------------------------------------------------------------------- /metrics/evaluation/masks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/README.md -------------------------------------------------------------------------------- /metrics/evaluation/masks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/.gitignore: -------------------------------------------------------------------------------- 1 | results -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/countless/README.md -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/countless2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/countless/countless2d.py -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/countless3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/countless/countless3d.py -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/images/gcim.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/countless/images/gcim.jpg -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/images/gray_segmentation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/countless/images/gray_segmentation.png -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/images/segmentation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/countless/images/segmentation.png -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/images/sparse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/countless/images/sparse.png -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/memprof/countless2d_gcim_N_1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/countless/memprof/countless2d_gcim_N_1000.png -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/memprof/countless2d_quick_gcim_N_1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/countless/memprof/countless2d_quick_gcim_N_1000.png -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/memprof/countless3d.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/countless/memprof/countless3d.png -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/memprof/countless3d_dynamic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/countless/memprof/countless3d_dynamic.png -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/memprof/countless3d_dynamic_generalized.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/countless/memprof/countless3d_dynamic_generalized.png -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/memprof/countless3d_generalized.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/countless/memprof/countless3d_generalized.png -------------------------------------------------------------------------------- /metrics/evaluation/masks/countless/requirements.txt: -------------------------------------------------------------------------------- 1 | Pillow>=6.2.0 2 | numpy>=1.16 3 | scipy 4 | tqdm 5 | memory_profiler 6 | six 7 | pytest -------------------------------------------------------------------------------- /metrics/evaluation/masks/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/masks/mask.py -------------------------------------------------------------------------------- /metrics/evaluation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/utils.py -------------------------------------------------------------------------------- /metrics/evaluation/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/metrics/evaluation/vis.py -------------------------------------------------------------------------------- /models/lpips_models/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/models/lpips_models/alex.pth -------------------------------------------------------------------------------- /models/lpips_models/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/models/lpips_models/squeeze.pth -------------------------------------------------------------------------------- /models/lpips_models/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/models/lpips_models/vgg.pth -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/requirements.txt -------------------------------------------------------------------------------- /streamlit_demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/README.md -------------------------------------------------------------------------------- /streamlit_demo/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/app.py -------------------------------------------------------------------------------- /streamlit_demo/demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/demo.png -------------------------------------------------------------------------------- /streamlit_demo/dnnlib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/dnnlib/__init__.py -------------------------------------------------------------------------------- /streamlit_demo/dnnlib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/dnnlib/util.py -------------------------------------------------------------------------------- /streamlit_demo/fcf_gan.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/fcf_gan.png -------------------------------------------------------------------------------- /streamlit_demo/legacy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/legacy.py -------------------------------------------------------------------------------- /streamlit_demo/packages.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/packages.txt -------------------------------------------------------------------------------- /streamlit_demo/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/requirements.txt -------------------------------------------------------------------------------- /streamlit_demo/setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/setup.sh -------------------------------------------------------------------------------- /streamlit_demo/test_512/church.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/test_512/church.jpeg -------------------------------------------------------------------------------- /streamlit_demo/test_512/door.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/test_512/door.jpeg -------------------------------------------------------------------------------- /streamlit_demo/test_512/floor.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/test_512/floor.jpeg -------------------------------------------------------------------------------- /streamlit_demo/test_512/house.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/test_512/house.jpeg -------------------------------------------------------------------------------- /streamlit_demo/test_512/person-cliff.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/test_512/person-cliff.jpeg -------------------------------------------------------------------------------- /streamlit_demo/test_512/person-fence.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/test_512/person-fence.png -------------------------------------------------------------------------------- /streamlit_demo/test_512/persons-white-fence.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/test_512/persons-white-fence.jpeg -------------------------------------------------------------------------------- /streamlit_demo/test_512/wall-1.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/test_512/wall-1.jpeg -------------------------------------------------------------------------------- /streamlit_demo/test_512/wall-2.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/test_512/wall-2.jpeg -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/__init__.py -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/custom_ops.py -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/misc.py -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/ops/__init__.py -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/ops/bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/ops/bias_act.cpp -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/ops/bias_act.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/ops/bias_act.cu -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/ops/bias_act.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/ops/bias_act.h -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/ops/bias_act.py -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/ops/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/ops/conv2d_gradfix.py -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/ops/conv2d_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/ops/conv2d_resample.py -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/ops/fma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/ops/fma.py -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/ops/grid_sample_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/ops/grid_sample_gradfix.py -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/ops/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/ops/upfirdn2d.cpp -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/ops/upfirdn2d.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/ops/upfirdn2d.cu -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/ops/upfirdn2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/ops/upfirdn2d.h -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/ops/upfirdn2d.py -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/persistence.py -------------------------------------------------------------------------------- /streamlit_demo/torch_utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/torch_utils/training_stats.py -------------------------------------------------------------------------------- /streamlit_demo/training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/__init__.py -------------------------------------------------------------------------------- /streamlit_demo/training/data/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/augment.py -------------------------------------------------------------------------------- /streamlit_demo/training/data/configs/medium_256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/configs/medium_256.yaml -------------------------------------------------------------------------------- /streamlit_demo/training/data/configs/medium_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/configs/medium_512.yaml -------------------------------------------------------------------------------- /streamlit_demo/training/data/configs/segm_256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/configs/segm_256.yaml -------------------------------------------------------------------------------- /streamlit_demo/training/data/configs/segm_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/configs/segm_512.yaml -------------------------------------------------------------------------------- /streamlit_demo/training/data/configs/thick_256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/configs/thick_256.yaml -------------------------------------------------------------------------------- /streamlit_demo/training/data/configs/thick_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/configs/thick_512.yaml -------------------------------------------------------------------------------- /streamlit_demo/training/data/configs/thin_256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/configs/thin_256.yaml -------------------------------------------------------------------------------- /streamlit_demo/training/data/configs/thin_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/configs/thin_512.yaml -------------------------------------------------------------------------------- /streamlit_demo/training/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/dataset.py -------------------------------------------------------------------------------- /streamlit_demo/training/data/demo_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/demo_loader.py -------------------------------------------------------------------------------- /streamlit_demo/training/data/gen_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/gen_loader.py -------------------------------------------------------------------------------- /streamlit_demo/training/data/lama_mask_generator_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/lama_mask_generator_test.py -------------------------------------------------------------------------------- /streamlit_demo/training/data/mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/mask_generator.py -------------------------------------------------------------------------------- /streamlit_demo/training/data/pred_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/data/pred_loader.py -------------------------------------------------------------------------------- /streamlit_demo/training/ffc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/ffc.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/__init__.py: -------------------------------------------------------------------------------- 1 | from .base import * -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/base.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/color150.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/color150.mat -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/mobilenet.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/object150_info.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/object150_info.csv -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/resnet.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib.zip -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/nn/__init__.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/__init__.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/batchnorm.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/comm.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/replicate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/replicate.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/tests/test_numeric_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/tests/test_numeric_batchnorm.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/tests/test_sync_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/tests/test_sync_batchnorm.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/unittest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/nn/modules/unittest.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/nn/parallel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/nn/parallel/__init__.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/nn/parallel/data_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/nn/parallel/data_parallel.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .th import * 2 | -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/utils/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/utils/data/__init__.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/utils/data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/utils/data/dataloader.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/utils/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/utils/data/dataset.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/utils/data/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/utils/data/distributed.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/utils/data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/utils/data/sampler.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/segm_lib/utils/th.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/segm_lib/utils/th.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/ade20k/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/ade20k/utils.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/high_receptive_pl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/high_receptive_pl.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/loss.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/perceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/perceptual.py -------------------------------------------------------------------------------- /streamlit_demo/training/losses/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/losses/vgg.py -------------------------------------------------------------------------------- /streamlit_demo/training/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/models.py -------------------------------------------------------------------------------- /streamlit_demo/training/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/networks.py -------------------------------------------------------------------------------- /streamlit_demo/training/training_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/streamlit_demo/training/training_loop.py -------------------------------------------------------------------------------- /tools/eval_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/tools/eval_sampler.py -------------------------------------------------------------------------------- /tools/eval_segm_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/tools/eval_segm_sampler.py -------------------------------------------------------------------------------- /tools/gen_masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/tools/gen_masks.py -------------------------------------------------------------------------------- /tools/gen_masks.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/tools/gen_masks.sh -------------------------------------------------------------------------------- /tools/gen_random_segm_masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/tools/gen_random_segm_masks.py -------------------------------------------------------------------------------- /tools/prepare_celebahq.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/tools/prepare_celebahq.sh -------------------------------------------------------------------------------- /tools/prepare_celebahq_evaluation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/tools/prepare_celebahq_evaluation.sh -------------------------------------------------------------------------------- /tools/prepare_places_evaluation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/tools/prepare_places_evaluation.sh -------------------------------------------------------------------------------- /tools/prepare_places_segm_evaluation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/tools/prepare_places_segm_evaluation.sh -------------------------------------------------------------------------------- /tools/prepare_places_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/tools/prepare_places_val.sh -------------------------------------------------------------------------------- /tools/train_shuffled.flist: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/tools/train_shuffled.flist -------------------------------------------------------------------------------- /tools/val_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/tools/val_sampler.py -------------------------------------------------------------------------------- /tools/val_shuffled.flist: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/tools/val_shuffled.flist -------------------------------------------------------------------------------- /torch_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/__init__.py -------------------------------------------------------------------------------- /torch_utils/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/custom_ops.py -------------------------------------------------------------------------------- /torch_utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/misc.py -------------------------------------------------------------------------------- /torch_utils/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/ops/__init__.py -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/ops/bias_act.cpp -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/ops/bias_act.cu -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/ops/bias_act.h -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/ops/bias_act.py -------------------------------------------------------------------------------- /torch_utils/ops/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/ops/conv2d_gradfix.py -------------------------------------------------------------------------------- /torch_utils/ops/conv2d_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/ops/conv2d_resample.py -------------------------------------------------------------------------------- /torch_utils/ops/fma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/ops/fma.py -------------------------------------------------------------------------------- /torch_utils/ops/grid_sample_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/ops/grid_sample_gradfix.py -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/ops/upfirdn2d.cpp -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/ops/upfirdn2d.cu -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/ops/upfirdn2d.h -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/ops/upfirdn2d.py -------------------------------------------------------------------------------- /torch_utils/persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/persistence.py -------------------------------------------------------------------------------- /torch_utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/torch_utils/training_stats.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/train.py -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/__init__.py -------------------------------------------------------------------------------- /training/data/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/augment.py -------------------------------------------------------------------------------- /training/data/configs/medium_256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/configs/medium_256.yaml -------------------------------------------------------------------------------- /training/data/configs/medium_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/configs/medium_512.yaml -------------------------------------------------------------------------------- /training/data/configs/segm_256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/configs/segm_256.yaml -------------------------------------------------------------------------------- /training/data/configs/segm_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/configs/segm_512.yaml -------------------------------------------------------------------------------- /training/data/configs/thick_256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/configs/thick_256.yaml -------------------------------------------------------------------------------- /training/data/configs/thick_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/configs/thick_512.yaml -------------------------------------------------------------------------------- /training/data/configs/thin_256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/configs/thin_256.yaml -------------------------------------------------------------------------------- /training/data/configs/thin_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/configs/thin_512.yaml -------------------------------------------------------------------------------- /training/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/dataset.py -------------------------------------------------------------------------------- /training/data/demo_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/demo_loader.py -------------------------------------------------------------------------------- /training/data/gen_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/gen_loader.py -------------------------------------------------------------------------------- /training/data/lama_mask_generator_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/lama_mask_generator_test.py -------------------------------------------------------------------------------- /training/data/mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/mask_generator.py -------------------------------------------------------------------------------- /training/data/pred_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/data/pred_loader.py -------------------------------------------------------------------------------- /training/ffc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/ffc.py -------------------------------------------------------------------------------- /training/losses/ade20k/__init__.py: -------------------------------------------------------------------------------- 1 | from .base import * -------------------------------------------------------------------------------- /training/losses/ade20k/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/base.py -------------------------------------------------------------------------------- /training/losses/ade20k/color150.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/color150.mat -------------------------------------------------------------------------------- /training/losses/ade20k/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/mobilenet.py -------------------------------------------------------------------------------- /training/losses/ade20k/object150_info.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/object150_info.csv -------------------------------------------------------------------------------- /training/losses/ade20k/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/resnet.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib.zip -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/nn/__init__.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/nn/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/nn/modules/__init__.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/nn/modules/batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/nn/modules/batchnorm.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/nn/modules/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/nn/modules/comm.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/nn/modules/replicate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/nn/modules/replicate.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/nn/modules/tests/test_numeric_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/nn/modules/tests/test_numeric_batchnorm.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/nn/modules/tests/test_sync_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/nn/modules/tests/test_sync_batchnorm.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/nn/modules/unittest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/nn/modules/unittest.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/nn/parallel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/nn/parallel/__init__.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/nn/parallel/data_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/nn/parallel/data_parallel.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .th import * 2 | -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/utils/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/utils/data/__init__.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/utils/data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/utils/data/dataloader.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/utils/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/utils/data/dataset.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/utils/data/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/utils/data/distributed.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/utils/data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/utils/data/sampler.py -------------------------------------------------------------------------------- /training/losses/ade20k/segm_lib/utils/th.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/segm_lib/utils/th.py -------------------------------------------------------------------------------- /training/losses/ade20k/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/ade20k/utils.py -------------------------------------------------------------------------------- /training/losses/high_receptive_pl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/high_receptive_pl.py -------------------------------------------------------------------------------- /training/losses/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/loss.py -------------------------------------------------------------------------------- /training/losses/perceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/perceptual.py -------------------------------------------------------------------------------- /training/losses/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/losses/vgg.py -------------------------------------------------------------------------------- /training/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/models.py -------------------------------------------------------------------------------- /training/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/networks.py -------------------------------------------------------------------------------- /training/training_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/training/training_loop.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .data_utils import * -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/FcF-Inpainting/HEAD/utils/data_utils.py --------------------------------------------------------------------------------