├── .gitignore ├── LICENSE ├── README.md ├── bin ├── analyze_errors.py ├── blur_predicts.py ├── calc_dataset_stats.py ├── debug │ └── analyze_overlapping_masks.sh ├── evaluate_predicts.py ├── evaluator_example.py ├── extract_masks.py ├── filter_sharded_dataset.py ├── gen_debug_mask_dataset.py ├── gen_mask_dataset.py ├── gen_mask_dataset_hydra.py ├── gen_outpainting_dataset.py ├── make_checkpoint.py ├── mask_example.py ├── paper_runfiles │ ├── blur_tests.sh │ ├── env.sh │ ├── find_best_checkpoint.py │ ├── generate_test_celeba-hq.sh │ ├── generate_test_ffhq.sh │ ├── generate_test_paris.sh │ ├── generate_test_paris_256.sh │ ├── generate_val_test.sh │ ├── predict_inner_features.sh │ └── update_test_data_stats.sh ├── predict.py ├── predict_inner_features.py ├── report_from_tb.py ├── sample_from_dataset.py ├── side_by_side.py ├── split_tar.py ├── to_jit.py └── train.py ├── colab └── LaMa_inpainting.ipynb ├── conda_env.yml ├── configs ├── analyze_mask_errors.yaml ├── data_gen │ ├── random_medium_256.yaml │ ├── random_medium_512.yaml │ ├── random_thick_256.yaml │ ├── random_thick_512.yaml │ ├── random_thin_256.yaml │ └── random_thin_512.yaml ├── debug_mask_gen.yaml ├── eval1.yaml ├── eval2.yaml ├── eval2_cpu.yaml ├── eval2_gpu.yaml ├── eval2_jpg.yaml ├── eval2_segm.yaml ├── eval2_segm_test.yaml ├── eval2_test.yaml ├── places2-categories_157.txt ├── prediction │ └── default.yaml ├── test_large_30k.lst └── training │ ├── ablv2_work.yaml │ ├── ablv2_work_ffc075.yaml │ ├── ablv2_work_md.yaml │ ├── ablv2_work_no_fm.yaml │ ├── ablv2_work_no_segmpl.yaml │ ├── ablv2_work_no_segmpl_csdilirpl.yaml │ ├── ablv2_work_no_segmpl_csdilirpl_celeba_csdilirpl1_new.yaml │ ├── ablv2_work_no_segmpl_csirpl.yaml │ ├── ablv2_work_no_segmpl_csirpl_celeba_csirpl03_new.yaml │ ├── ablv2_work_no_segmpl_vgg.yaml │ ├── ablv2_work_no_segmpl_vgg_celeba_l2_vgg003_new.yaml │ ├── ablv2_work_nodil_segmpl.yaml │ ├── ablv2_work_small_holes.yaml │ ├── big-lama-celeba.yaml │ ├── big-lama-regular-celeba.yaml │ ├── big-lama-regular.yaml │ ├── big-lama.yaml │ ├── data │ ├── abl-02-thin-bb.yaml │ ├── abl-04-256-mh-dist-celeba.yaml │ ├── abl-04-256-mh-dist-web.yaml │ └── abl-04-256-mh-dist.yaml │ ├── discriminator │ └── pix2pixhd_nlayer.yaml │ ├── evaluator │ └── default_inpainted.yaml │ ├── generator │ ├── ffc_resnet_075.yaml │ ├── pix2pixhd_global.yaml │ ├── pix2pixhd_global_sigmoid.yaml │ └── pix2pixhd_multidilated_catin_4dil_9b.yaml │ ├── hydra │ ├── no_time.yaml │ └── overrides.yaml │ ├── lama-fourier-celeba.yaml │ ├── lama-fourier.yaml │ ├── lama-regular-celeba.yaml │ ├── lama-regular.yaml │ ├── lama_small_train_masks.yaml │ ├── location │ ├── celeba_example.yaml │ ├── docker.yaml │ └── places_example.yaml │ ├── optimizers │ └── default_optimizers.yaml │ ├── trainer │ ├── any_gpu_large_ssim_ddp_final.yaml │ ├── any_gpu_large_ssim_ddp_final_benchmark.yaml │ └── any_gpu_large_ssim_ddp_final_celeba.yaml │ └── visualizer │ └── directory.yaml ├── docker ├── 1_generate_masks_from_raw_images.sh ├── 2_predict_with_gpu.sh ├── 3_evaluate.sh ├── Dockerfile ├── Dockerfile-cuda111 ├── build-cuda111.sh ├── build.sh └── entrypoint.sh ├── fetch_data ├── celebahq_dataset_prepare.sh ├── celebahq_gen_masks.sh ├── eval_sampler.py ├── places_challenge_train_download.sh ├── places_standard_evaluation_prepare_data.sh ├── places_standard_test_val_gen_masks.sh ├── places_standard_test_val_prepare.sh ├── places_standard_test_val_sample.sh ├── places_standard_train_prepare.sh ├── sampler.py ├── train_shuffled.flist └── val_shuffled.flist ├── models ├── ade20k │ ├── __init__.py │ ├── base.py │ ├── color150.mat │ ├── mobilenet.py │ ├── object150_info.csv │ ├── resnet.py │ ├── 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 └── lpips_models │ ├── alex.pth │ ├── squeeze.pth │ └── vgg.pth ├── requirements.txt └── saicinpainting ├── __init__.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 │ │ └── test.py │ └── mask.py ├── refinement.py ├── utils.py └── vis.py ├── training ├── __init__.py ├── data │ ├── __init__.py │ ├── aug.py │ ├── datasets.py │ └── masks.py ├── losses │ ├── __init__.py │ ├── adversarial.py │ ├── constants.py │ ├── distance_weighting.py │ ├── feature_matching.py │ ├── perceptual.py │ ├── segmentation.py │ └── style_loss.py ├── modules │ ├── __init__.py │ ├── base.py │ ├── depthwise_sep_conv.py │ ├── fake_fakes.py │ ├── ffc.py │ ├── multidilated_conv.py │ ├── multiscale.py │ ├── pix2pixhd.py │ ├── spatial_transform.py │ └── squeeze_excitation.py ├── trainers │ ├── __init__.py │ ├── base.py │ └── default.py └── visualizers │ ├── __init__.py │ ├── base.py │ ├── colors.py │ ├── directory.py │ └── noop.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/README.md -------------------------------------------------------------------------------- /bin/analyze_errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/analyze_errors.py -------------------------------------------------------------------------------- /bin/blur_predicts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/blur_predicts.py -------------------------------------------------------------------------------- /bin/calc_dataset_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/calc_dataset_stats.py -------------------------------------------------------------------------------- /bin/debug/analyze_overlapping_masks.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/debug/analyze_overlapping_masks.sh -------------------------------------------------------------------------------- /bin/evaluate_predicts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/evaluate_predicts.py -------------------------------------------------------------------------------- /bin/evaluator_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/evaluator_example.py -------------------------------------------------------------------------------- /bin/extract_masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/extract_masks.py -------------------------------------------------------------------------------- /bin/filter_sharded_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/filter_sharded_dataset.py -------------------------------------------------------------------------------- /bin/gen_debug_mask_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/gen_debug_mask_dataset.py -------------------------------------------------------------------------------- /bin/gen_mask_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/gen_mask_dataset.py -------------------------------------------------------------------------------- /bin/gen_mask_dataset_hydra.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/gen_mask_dataset_hydra.py -------------------------------------------------------------------------------- /bin/gen_outpainting_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/gen_outpainting_dataset.py -------------------------------------------------------------------------------- /bin/make_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/make_checkpoint.py -------------------------------------------------------------------------------- /bin/mask_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/mask_example.py -------------------------------------------------------------------------------- /bin/paper_runfiles/blur_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/paper_runfiles/blur_tests.sh -------------------------------------------------------------------------------- /bin/paper_runfiles/env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/paper_runfiles/env.sh -------------------------------------------------------------------------------- /bin/paper_runfiles/find_best_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/paper_runfiles/find_best_checkpoint.py -------------------------------------------------------------------------------- /bin/paper_runfiles/generate_test_celeba-hq.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/paper_runfiles/generate_test_celeba-hq.sh -------------------------------------------------------------------------------- /bin/paper_runfiles/generate_test_ffhq.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/paper_runfiles/generate_test_ffhq.sh -------------------------------------------------------------------------------- /bin/paper_runfiles/generate_test_paris.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/paper_runfiles/generate_test_paris.sh -------------------------------------------------------------------------------- /bin/paper_runfiles/generate_test_paris_256.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/paper_runfiles/generate_test_paris_256.sh -------------------------------------------------------------------------------- /bin/paper_runfiles/generate_val_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/paper_runfiles/generate_val_test.sh -------------------------------------------------------------------------------- /bin/paper_runfiles/predict_inner_features.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/paper_runfiles/predict_inner_features.sh -------------------------------------------------------------------------------- /bin/paper_runfiles/update_test_data_stats.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/paper_runfiles/update_test_data_stats.sh -------------------------------------------------------------------------------- /bin/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/predict.py -------------------------------------------------------------------------------- /bin/predict_inner_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/predict_inner_features.py -------------------------------------------------------------------------------- /bin/report_from_tb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/report_from_tb.py -------------------------------------------------------------------------------- /bin/sample_from_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/sample_from_dataset.py -------------------------------------------------------------------------------- /bin/side_by_side.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/side_by_side.py -------------------------------------------------------------------------------- /bin/split_tar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/split_tar.py -------------------------------------------------------------------------------- /bin/to_jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/to_jit.py -------------------------------------------------------------------------------- /bin/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/bin/train.py -------------------------------------------------------------------------------- /colab/LaMa_inpainting.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/colab/LaMa_inpainting.ipynb -------------------------------------------------------------------------------- /conda_env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/conda_env.yml -------------------------------------------------------------------------------- /configs/analyze_mask_errors.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/analyze_mask_errors.yaml -------------------------------------------------------------------------------- /configs/data_gen/random_medium_256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/data_gen/random_medium_256.yaml -------------------------------------------------------------------------------- /configs/data_gen/random_medium_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/data_gen/random_medium_512.yaml -------------------------------------------------------------------------------- /configs/data_gen/random_thick_256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/data_gen/random_thick_256.yaml -------------------------------------------------------------------------------- /configs/data_gen/random_thick_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/data_gen/random_thick_512.yaml -------------------------------------------------------------------------------- /configs/data_gen/random_thin_256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/data_gen/random_thin_256.yaml -------------------------------------------------------------------------------- /configs/data_gen/random_thin_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/data_gen/random_thin_512.yaml -------------------------------------------------------------------------------- /configs/debug_mask_gen.yaml: -------------------------------------------------------------------------------- 1 | img_ext: .jpg 2 | 3 | gen_kwargs: 4 | mask_size: 200 5 | step: 0.5 6 | -------------------------------------------------------------------------------- /configs/eval1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/eval1.yaml -------------------------------------------------------------------------------- /configs/eval2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/eval2.yaml -------------------------------------------------------------------------------- /configs/eval2_cpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/eval2_cpu.yaml -------------------------------------------------------------------------------- /configs/eval2_gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/eval2_gpu.yaml -------------------------------------------------------------------------------- /configs/eval2_jpg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/eval2_jpg.yaml -------------------------------------------------------------------------------- /configs/eval2_segm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/eval2_segm.yaml -------------------------------------------------------------------------------- /configs/eval2_segm_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/eval2_segm_test.yaml -------------------------------------------------------------------------------- /configs/eval2_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/eval2_test.yaml -------------------------------------------------------------------------------- /configs/places2-categories_157.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/places2-categories_157.txt -------------------------------------------------------------------------------- /configs/prediction/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/prediction/default.yaml -------------------------------------------------------------------------------- /configs/test_large_30k.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/test_large_30k.lst -------------------------------------------------------------------------------- /configs/training/ablv2_work.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/ablv2_work.yaml -------------------------------------------------------------------------------- /configs/training/ablv2_work_ffc075.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/ablv2_work_ffc075.yaml -------------------------------------------------------------------------------- /configs/training/ablv2_work_md.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/ablv2_work_md.yaml -------------------------------------------------------------------------------- /configs/training/ablv2_work_no_fm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/ablv2_work_no_fm.yaml -------------------------------------------------------------------------------- /configs/training/ablv2_work_no_segmpl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/ablv2_work_no_segmpl.yaml -------------------------------------------------------------------------------- /configs/training/ablv2_work_no_segmpl_csdilirpl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/ablv2_work_no_segmpl_csdilirpl.yaml -------------------------------------------------------------------------------- /configs/training/ablv2_work_no_segmpl_csdilirpl_celeba_csdilirpl1_new.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/ablv2_work_no_segmpl_csdilirpl_celeba_csdilirpl1_new.yaml -------------------------------------------------------------------------------- /configs/training/ablv2_work_no_segmpl_csirpl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/ablv2_work_no_segmpl_csirpl.yaml -------------------------------------------------------------------------------- /configs/training/ablv2_work_no_segmpl_csirpl_celeba_csirpl03_new.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/ablv2_work_no_segmpl_csirpl_celeba_csirpl03_new.yaml -------------------------------------------------------------------------------- /configs/training/ablv2_work_no_segmpl_vgg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/ablv2_work_no_segmpl_vgg.yaml -------------------------------------------------------------------------------- /configs/training/ablv2_work_no_segmpl_vgg_celeba_l2_vgg003_new.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/ablv2_work_no_segmpl_vgg_celeba_l2_vgg003_new.yaml -------------------------------------------------------------------------------- /configs/training/ablv2_work_nodil_segmpl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/ablv2_work_nodil_segmpl.yaml -------------------------------------------------------------------------------- /configs/training/ablv2_work_small_holes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/ablv2_work_small_holes.yaml -------------------------------------------------------------------------------- /configs/training/big-lama-celeba.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/big-lama-celeba.yaml -------------------------------------------------------------------------------- /configs/training/big-lama-regular-celeba.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/big-lama-regular-celeba.yaml -------------------------------------------------------------------------------- /configs/training/big-lama-regular.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/big-lama-regular.yaml -------------------------------------------------------------------------------- /configs/training/big-lama.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/big-lama.yaml -------------------------------------------------------------------------------- /configs/training/data/abl-02-thin-bb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/data/abl-02-thin-bb.yaml -------------------------------------------------------------------------------- /configs/training/data/abl-04-256-mh-dist-celeba.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/data/abl-04-256-mh-dist-celeba.yaml -------------------------------------------------------------------------------- /configs/training/data/abl-04-256-mh-dist-web.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/data/abl-04-256-mh-dist-web.yaml -------------------------------------------------------------------------------- /configs/training/data/abl-04-256-mh-dist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/data/abl-04-256-mh-dist.yaml -------------------------------------------------------------------------------- /configs/training/discriminator/pix2pixhd_nlayer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/discriminator/pix2pixhd_nlayer.yaml -------------------------------------------------------------------------------- /configs/training/evaluator/default_inpainted.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/evaluator/default_inpainted.yaml -------------------------------------------------------------------------------- /configs/training/generator/ffc_resnet_075.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/generator/ffc_resnet_075.yaml -------------------------------------------------------------------------------- /configs/training/generator/pix2pixhd_global.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/generator/pix2pixhd_global.yaml -------------------------------------------------------------------------------- /configs/training/generator/pix2pixhd_global_sigmoid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/generator/pix2pixhd_global_sigmoid.yaml -------------------------------------------------------------------------------- /configs/training/generator/pix2pixhd_multidilated_catin_4dil_9b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/generator/pix2pixhd_multidilated_catin_4dil_9b.yaml -------------------------------------------------------------------------------- /configs/training/hydra/no_time.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/hydra/no_time.yaml -------------------------------------------------------------------------------- /configs/training/hydra/overrides.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/hydra/overrides.yaml -------------------------------------------------------------------------------- /configs/training/lama-fourier-celeba.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/lama-fourier-celeba.yaml -------------------------------------------------------------------------------- /configs/training/lama-fourier.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/lama-fourier.yaml -------------------------------------------------------------------------------- /configs/training/lama-regular-celeba.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/lama-regular-celeba.yaml -------------------------------------------------------------------------------- /configs/training/lama-regular.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/lama-regular.yaml -------------------------------------------------------------------------------- /configs/training/lama_small_train_masks.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/lama_small_train_masks.yaml -------------------------------------------------------------------------------- /configs/training/location/celeba_example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/location/celeba_example.yaml -------------------------------------------------------------------------------- /configs/training/location/docker.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/location/docker.yaml -------------------------------------------------------------------------------- /configs/training/location/places_example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/location/places_example.yaml -------------------------------------------------------------------------------- /configs/training/optimizers/default_optimizers.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/optimizers/default_optimizers.yaml -------------------------------------------------------------------------------- /configs/training/trainer/any_gpu_large_ssim_ddp_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/trainer/any_gpu_large_ssim_ddp_final.yaml -------------------------------------------------------------------------------- /configs/training/trainer/any_gpu_large_ssim_ddp_final_benchmark.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/trainer/any_gpu_large_ssim_ddp_final_benchmark.yaml -------------------------------------------------------------------------------- /configs/training/trainer/any_gpu_large_ssim_ddp_final_celeba.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/trainer/any_gpu_large_ssim_ddp_final_celeba.yaml -------------------------------------------------------------------------------- /configs/training/visualizer/directory.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/configs/training/visualizer/directory.yaml -------------------------------------------------------------------------------- /docker/1_generate_masks_from_raw_images.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/docker/1_generate_masks_from_raw_images.sh -------------------------------------------------------------------------------- /docker/2_predict_with_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/docker/2_predict_with_gpu.sh -------------------------------------------------------------------------------- /docker/3_evaluate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/docker/3_evaluate.sh -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/Dockerfile-cuda111: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/docker/Dockerfile-cuda111 -------------------------------------------------------------------------------- /docker/build-cuda111.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/docker/build-cuda111.sh -------------------------------------------------------------------------------- /docker/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/docker/build.sh -------------------------------------------------------------------------------- /docker/entrypoint.sh: -------------------------------------------------------------------------------- 1 | #!/bin/bash 2 | 3 | exec $@ 4 | -------------------------------------------------------------------------------- /fetch_data/celebahq_dataset_prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/fetch_data/celebahq_dataset_prepare.sh -------------------------------------------------------------------------------- /fetch_data/celebahq_gen_masks.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/fetch_data/celebahq_gen_masks.sh -------------------------------------------------------------------------------- /fetch_data/eval_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/fetch_data/eval_sampler.py -------------------------------------------------------------------------------- /fetch_data/places_challenge_train_download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/fetch_data/places_challenge_train_download.sh -------------------------------------------------------------------------------- /fetch_data/places_standard_evaluation_prepare_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/fetch_data/places_standard_evaluation_prepare_data.sh -------------------------------------------------------------------------------- /fetch_data/places_standard_test_val_gen_masks.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/fetch_data/places_standard_test_val_gen_masks.sh -------------------------------------------------------------------------------- /fetch_data/places_standard_test_val_prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/fetch_data/places_standard_test_val_prepare.sh -------------------------------------------------------------------------------- /fetch_data/places_standard_test_val_sample.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/fetch_data/places_standard_test_val_sample.sh -------------------------------------------------------------------------------- /fetch_data/places_standard_train_prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/fetch_data/places_standard_train_prepare.sh -------------------------------------------------------------------------------- /fetch_data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/fetch_data/sampler.py -------------------------------------------------------------------------------- /fetch_data/train_shuffled.flist: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/fetch_data/train_shuffled.flist -------------------------------------------------------------------------------- /fetch_data/val_shuffled.flist: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/fetch_data/val_shuffled.flist -------------------------------------------------------------------------------- /models/ade20k/__init__.py: -------------------------------------------------------------------------------- 1 | from .base import * -------------------------------------------------------------------------------- /models/ade20k/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/base.py -------------------------------------------------------------------------------- /models/ade20k/color150.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/color150.mat -------------------------------------------------------------------------------- /models/ade20k/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/mobilenet.py -------------------------------------------------------------------------------- /models/ade20k/object150_info.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/object150_info.csv -------------------------------------------------------------------------------- /models/ade20k/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/resnet.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/nn/__init__.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/nn/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/nn/modules/__init__.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/nn/modules/batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/nn/modules/batchnorm.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/nn/modules/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/nn/modules/comm.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/nn/modules/replicate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/nn/modules/replicate.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/nn/modules/tests/test_numeric_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/nn/modules/tests/test_numeric_batchnorm.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/nn/modules/tests/test_sync_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/nn/modules/tests/test_sync_batchnorm.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/nn/modules/unittest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/nn/modules/unittest.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/nn/parallel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/nn/parallel/__init__.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/nn/parallel/data_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/nn/parallel/data_parallel.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .th import * 2 | -------------------------------------------------------------------------------- /models/ade20k/segm_lib/utils/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/utils/data/__init__.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/utils/data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/utils/data/dataloader.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/utils/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/utils/data/dataset.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/utils/data/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/utils/data/distributed.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/utils/data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/utils/data/sampler.py -------------------------------------------------------------------------------- /models/ade20k/segm_lib/utils/th.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/segm_lib/utils/th.py -------------------------------------------------------------------------------- /models/ade20k/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/ade20k/utils.py -------------------------------------------------------------------------------- /models/lpips_models/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/lpips_models/alex.pth -------------------------------------------------------------------------------- /models/lpips_models/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/lpips_models/squeeze.pth -------------------------------------------------------------------------------- /models/lpips_models/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/models/lpips_models/vgg.pth -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/requirements.txt -------------------------------------------------------------------------------- /saicinpainting/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /saicinpainting/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/__init__.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/data.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/evaluator.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /saicinpainting/evaluation/losses/base_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/losses/base_loss.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/losses/fid/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /saicinpainting/evaluation/losses/fid/fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/losses/fid/fid_score.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/losses/fid/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/losses/fid/inception.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/losses/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/losses/lpips.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/losses/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/losses/ssim.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/README.md -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/.gitignore: -------------------------------------------------------------------------------- 1 | results -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/README.md -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/countless2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/countless2d.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/countless3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/countless3d.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/images/gcim.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/images/gcim.jpg -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/images/gray_segmentation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/images/gray_segmentation.png -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/images/segmentation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/images/segmentation.png -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/images/sparse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/images/sparse.png -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/memprof/countless2d_gcim_N_1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/memprof/countless2d_gcim_N_1000.png -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/memprof/countless2d_quick_gcim_N_1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/memprof/countless2d_quick_gcim_N_1000.png -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/memprof/countless3d.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/memprof/countless3d.png -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/memprof/countless3d_dynamic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/memprof/countless3d_dynamic.png -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/memprof/countless3d_dynamic_generalized.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/memprof/countless3d_dynamic_generalized.png -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/memprof/countless3d_generalized.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/memprof/countless3d_generalized.png -------------------------------------------------------------------------------- /saicinpainting/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 -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/countless/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/countless/test.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/masks/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/masks/mask.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/refinement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/refinement.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/utils.py -------------------------------------------------------------------------------- /saicinpainting/evaluation/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/evaluation/vis.py -------------------------------------------------------------------------------- /saicinpainting/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /saicinpainting/training/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /saicinpainting/training/data/aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/data/aug.py -------------------------------------------------------------------------------- /saicinpainting/training/data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/data/datasets.py -------------------------------------------------------------------------------- /saicinpainting/training/data/masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/data/masks.py -------------------------------------------------------------------------------- /saicinpainting/training/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /saicinpainting/training/losses/adversarial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/losses/adversarial.py -------------------------------------------------------------------------------- /saicinpainting/training/losses/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/losses/constants.py -------------------------------------------------------------------------------- /saicinpainting/training/losses/distance_weighting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/losses/distance_weighting.py -------------------------------------------------------------------------------- /saicinpainting/training/losses/feature_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/losses/feature_matching.py -------------------------------------------------------------------------------- /saicinpainting/training/losses/perceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/losses/perceptual.py -------------------------------------------------------------------------------- /saicinpainting/training/losses/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/losses/segmentation.py -------------------------------------------------------------------------------- /saicinpainting/training/losses/style_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/losses/style_loss.py -------------------------------------------------------------------------------- /saicinpainting/training/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/modules/__init__.py -------------------------------------------------------------------------------- /saicinpainting/training/modules/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/modules/base.py -------------------------------------------------------------------------------- /saicinpainting/training/modules/depthwise_sep_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/modules/depthwise_sep_conv.py -------------------------------------------------------------------------------- /saicinpainting/training/modules/fake_fakes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/modules/fake_fakes.py -------------------------------------------------------------------------------- /saicinpainting/training/modules/ffc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/modules/ffc.py -------------------------------------------------------------------------------- /saicinpainting/training/modules/multidilated_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/modules/multidilated_conv.py -------------------------------------------------------------------------------- /saicinpainting/training/modules/multiscale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/modules/multiscale.py -------------------------------------------------------------------------------- /saicinpainting/training/modules/pix2pixhd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/modules/pix2pixhd.py -------------------------------------------------------------------------------- /saicinpainting/training/modules/spatial_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/modules/spatial_transform.py -------------------------------------------------------------------------------- /saicinpainting/training/modules/squeeze_excitation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/modules/squeeze_excitation.py -------------------------------------------------------------------------------- /saicinpainting/training/trainers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/trainers/__init__.py -------------------------------------------------------------------------------- /saicinpainting/training/trainers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/trainers/base.py -------------------------------------------------------------------------------- /saicinpainting/training/trainers/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/trainers/default.py -------------------------------------------------------------------------------- /saicinpainting/training/visualizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/visualizers/__init__.py -------------------------------------------------------------------------------- /saicinpainting/training/visualizers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/visualizers/base.py -------------------------------------------------------------------------------- /saicinpainting/training/visualizers/colors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/visualizers/colors.py -------------------------------------------------------------------------------- /saicinpainting/training/visualizers/directory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/visualizers/directory.py -------------------------------------------------------------------------------- /saicinpainting/training/visualizers/noop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/training/visualizers/noop.py -------------------------------------------------------------------------------- /saicinpainting/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/geomagical/lama-with-refiner/HEAD/saicinpainting/utils.py --------------------------------------------------------------------------------