├── .gitignore ├── LICENSE ├── README.md ├── VERSION ├── assets ├── rsfnet-map.mp4 ├── rsfnet-palette.mp4 ├── rsfnet-saliency.mp4 └── rsfnet-semantic.mp4 ├── basicsr ├── __init__.py ├── archs │ ├── __init__.py │ ├── arch_util.py │ ├── renderer_arch.py │ ├── rsfnet_arch.py │ └── vgg_arch.py ├── backbones │ ├── __init__.py │ ├── resnet_backbone.py │ ├── resnext_backbone.py │ └── vgg_backbone.py ├── data │ ├── __init__.py │ ├── data_sampler.py │ ├── data_util.py │ ├── paired_image_dataset.py │ ├── prefetch_dataloader.py │ ├── rsfnet_dataset.py │ ├── single_image_dataset.py │ ├── transforms.py │ └── unpaired_image_dataset.py ├── heads │ ├── __init__.py │ ├── mask_head.py │ ├── paramrecolor_head.py │ └── segrecolor_head.py ├── losses │ ├── __init__.py │ ├── loss_util.py │ └── losses.py ├── metrics │ ├── README.md │ ├── README_CN.md │ ├── __init__.py │ ├── fid.py │ ├── lpips.py │ ├── metric_util.py │ ├── niqe.py │ ├── niqe_pris_params.npz │ ├── psnr_ssim.py │ └── test_metrics │ │ └── test_psnr_ssim.py ├── models │ ├── __init__.py │ ├── base_model.py │ ├── lr_scheduler.py │ ├── rsfnet_maskin_model.py │ └── rsfnet_model.py ├── necks │ ├── __init__.py │ └── fpn_neck.py ├── ops │ ├── __init__.py │ └── color │ │ ├── __init__.py │ │ ├── pyinterfaces.py │ │ └── src │ │ ├── color_cuda.cpp │ │ ├── color_kernel.cu │ │ └── setup.py ├── test.py ├── train.py ├── utils │ ├── __init__.py │ ├── color_util.py │ ├── conv_util.py │ ├── diffjpeg.py │ ├── dist_util.py │ ├── download_util.py │ ├── file_client.py │ ├── flow_util.py │ ├── img_process_util.py │ ├── img_util.py │ ├── lmdb_util.py │ ├── logger.py │ ├── matlab_functions.py │ ├── misc.py │ ├── options.py │ ├── registry.py │ ├── visualization.py │ ├── visualizer │ │ ├── __init__.py │ │ ├── html.py │ │ ├── util.py │ │ └── visualizer.py │ └── weight_init.py └── version.py ├── datasets └── generates │ ├── PaletteGenerate │ ├── GUI.py │ ├── README.md │ ├── __init__.py │ ├── harmonization.py │ ├── main.py │ ├── palette.py │ ├── palette_generate.py │ ├── transfer.py │ └── util.py │ ├── SaliencyDetect │ ├── __init__.py │ ├── networks │ │ ├── __init__.py │ │ ├── deeplab_resnet.py │ │ ├── joint_poolnet.py │ │ ├── poolnet.py │ │ └── vgg.py │ └── saliency_detect.py │ ├── palette_mask_generate.py │ └── saliency_mask_generate.py ├── options ├── infer │ ├── infer_rsfnet_global_fivek.yml │ ├── infer_rsfnet_map_fivek.yml │ ├── infer_rsfnet_map_ppr10k_a.yml │ ├── infer_rsfnet_map_ppr10k_b.yml │ ├── infer_rsfnet_map_ppr10k_c.yml │ ├── infer_rsfnet_palette_fivek.yml │ ├── infer_rsfnet_saliency_fivek.yml │ └── infer_rsfnet_saliency_ppr10k.yml ├── test │ └── RSFNet │ │ ├── test_rsfnet_global_fivek_zas.yml │ │ ├── test_rsfnet_global_fivek_zwc.yml │ │ ├── test_rsfnet_map_fivek_zas.yml │ │ ├── test_rsfnet_map_fivek_zwc.yml │ │ ├── test_rsfnet_map_ppr10k_a.yml │ │ ├── test_rsfnet_map_ppr10k_b.yml │ │ ├── test_rsfnet_map_ppr10k_c.yml │ │ ├── test_rsfnet_maskin_palette_fivek_zas.yml │ │ ├── test_rsfnet_maskin_saliency_fivek_zas.yml │ │ ├── test_rsfnet_palette_fivek_zas.yml │ │ ├── test_rsfnet_palette_fivek_zwc.yml │ │ ├── test_rsfnet_saliency_fivek_zas.yml │ │ ├── test_rsfnet_saliency_fivek_zwc.yml │ │ ├── test_rsfnet_saliency_ppr10k_a.yml │ │ ├── test_rsfnet_saliency_ppr10k_b.yml │ │ ├── test_rsfnet_saliency_ppr10k_c.yml │ │ ├── test_rsfnet_semantic_fivek_zas.yml │ │ └── test_rsfnet_semantic_fivek_zwc.yml └── train │ └── RSFNet │ ├── train_rsfnet_global_fivek_zas.yml │ ├── train_rsfnet_global_fivek_zwc.yml │ ├── train_rsfnet_map_fivek_zas.yml │ ├── train_rsfnet_map_fivek_zwc.yml │ ├── train_rsfnet_map_ppr10k_a.yml │ ├── train_rsfnet_map_ppr10k_b.yml │ ├── train_rsfnet_map_ppr10k_c.yml │ ├── train_rsfnet_maskin_palette_fivek_zas.yml │ ├── train_rsfnet_maskin_saliency_fivek_zas.yml │ ├── train_rsfnet_palette_fivek_zas.yml │ ├── train_rsfnet_palette_fivek_zwc.yml │ ├── train_rsfnet_saliency_fivek_zas.yml │ ├── train_rsfnet_saliency_fivek_zwc.yml │ ├── train_rsfnet_saliency_ppr10k_a.yml │ ├── train_rsfnet_saliency_ppr10k_b.yml │ ├── train_rsfnet_saliency_ppr10k_c.yml │ ├── train_rsfnet_semantic_fivek_zas.yml │ └── train_rsfnet_semantic_fivek_zwc.yml └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/README.md -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 1.3.5 2 | -------------------------------------------------------------------------------- /assets/rsfnet-map.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/assets/rsfnet-map.mp4 -------------------------------------------------------------------------------- /assets/rsfnet-palette.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/assets/rsfnet-palette.mp4 -------------------------------------------------------------------------------- /assets/rsfnet-saliency.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/assets/rsfnet-saliency.mp4 -------------------------------------------------------------------------------- /assets/rsfnet-semantic.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/assets/rsfnet-semantic.mp4 -------------------------------------------------------------------------------- /basicsr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/archs/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/archs/arch_util.py -------------------------------------------------------------------------------- /basicsr/archs/renderer_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/archs/renderer_arch.py -------------------------------------------------------------------------------- /basicsr/archs/rsfnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/archs/rsfnet_arch.py -------------------------------------------------------------------------------- /basicsr/archs/vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/archs/vgg_arch.py -------------------------------------------------------------------------------- /basicsr/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/backbones/__init__.py -------------------------------------------------------------------------------- /basicsr/backbones/resnet_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/backbones/resnet_backbone.py -------------------------------------------------------------------------------- /basicsr/backbones/resnext_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/backbones/resnext_backbone.py -------------------------------------------------------------------------------- /basicsr/backbones/vgg_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/backbones/vgg_backbone.py -------------------------------------------------------------------------------- /basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/data/__init__.py -------------------------------------------------------------------------------- /basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/data/data_util.py -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /basicsr/data/rsfnet_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/data/rsfnet_dataset.py -------------------------------------------------------------------------------- /basicsr/data/single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/data/single_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/data/transforms.py -------------------------------------------------------------------------------- /basicsr/data/unpaired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/data/unpaired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/heads/__init__.py -------------------------------------------------------------------------------- /basicsr/heads/mask_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/heads/mask_head.py -------------------------------------------------------------------------------- /basicsr/heads/paramrecolor_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/heads/paramrecolor_head.py -------------------------------------------------------------------------------- /basicsr/heads/segrecolor_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/heads/segrecolor_head.py -------------------------------------------------------------------------------- /basicsr/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/losses/__init__.py -------------------------------------------------------------------------------- /basicsr/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/losses/loss_util.py -------------------------------------------------------------------------------- /basicsr/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/losses/losses.py -------------------------------------------------------------------------------- /basicsr/metrics/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/metrics/README.md -------------------------------------------------------------------------------- /basicsr/metrics/README_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/metrics/README_CN.md -------------------------------------------------------------------------------- /basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /basicsr/metrics/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/metrics/fid.py -------------------------------------------------------------------------------- /basicsr/metrics/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/metrics/lpips.py -------------------------------------------------------------------------------- /basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/metrics/niqe.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe_pris_params.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/metrics/niqe_pris_params.npz -------------------------------------------------------------------------------- /basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/metrics/test_metrics/test_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/metrics/test_metrics/test_psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/models/__init__.py -------------------------------------------------------------------------------- /basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/models/base_model.py -------------------------------------------------------------------------------- /basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /basicsr/models/rsfnet_maskin_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/models/rsfnet_maskin_model.py -------------------------------------------------------------------------------- /basicsr/models/rsfnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/models/rsfnet_model.py -------------------------------------------------------------------------------- /basicsr/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/necks/__init__.py -------------------------------------------------------------------------------- /basicsr/necks/fpn_neck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/necks/fpn_neck.py -------------------------------------------------------------------------------- /basicsr/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /basicsr/ops/color/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/ops/color/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/color/pyinterfaces.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/ops/color/pyinterfaces.py -------------------------------------------------------------------------------- /basicsr/ops/color/src/color_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/ops/color/src/color_cuda.cpp -------------------------------------------------------------------------------- /basicsr/ops/color/src/color_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/ops/color/src/color_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/color/src/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/ops/color/src/setup.py -------------------------------------------------------------------------------- /basicsr/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/test.py -------------------------------------------------------------------------------- /basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/train.py -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/color_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/color_util.py -------------------------------------------------------------------------------- /basicsr/utils/conv_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/conv_util.py -------------------------------------------------------------------------------- /basicsr/utils/diffjpeg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/diffjpeg.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/download_util.py -------------------------------------------------------------------------------- /basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /basicsr/utils/flow_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/flow_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_process_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/img_process_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /basicsr/utils/lmdb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/lmdb_util.py -------------------------------------------------------------------------------- /basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/logger.py -------------------------------------------------------------------------------- /basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/options.py -------------------------------------------------------------------------------- /basicsr/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/registry.py -------------------------------------------------------------------------------- /basicsr/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/visualization.py -------------------------------------------------------------------------------- /basicsr/utils/visualizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/visualizer/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/visualizer/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/visualizer/html.py -------------------------------------------------------------------------------- /basicsr/utils/visualizer/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/visualizer/util.py -------------------------------------------------------------------------------- /basicsr/utils/visualizer/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/visualizer/visualizer.py -------------------------------------------------------------------------------- /basicsr/utils/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/utils/weight_init.py -------------------------------------------------------------------------------- /basicsr/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/basicsr/version.py -------------------------------------------------------------------------------- /datasets/generates/PaletteGenerate/GUI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/PaletteGenerate/GUI.py -------------------------------------------------------------------------------- /datasets/generates/PaletteGenerate/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/PaletteGenerate/README.md -------------------------------------------------------------------------------- /datasets/generates/PaletteGenerate/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/generates/PaletteGenerate/harmonization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/PaletteGenerate/harmonization.py -------------------------------------------------------------------------------- /datasets/generates/PaletteGenerate/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/PaletteGenerate/main.py -------------------------------------------------------------------------------- /datasets/generates/PaletteGenerate/palette.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/PaletteGenerate/palette.py -------------------------------------------------------------------------------- /datasets/generates/PaletteGenerate/palette_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/PaletteGenerate/palette_generate.py -------------------------------------------------------------------------------- /datasets/generates/PaletteGenerate/transfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/PaletteGenerate/transfer.py -------------------------------------------------------------------------------- /datasets/generates/PaletteGenerate/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/PaletteGenerate/util.py -------------------------------------------------------------------------------- /datasets/generates/SaliencyDetect/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/generates/SaliencyDetect/networks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/generates/SaliencyDetect/networks/deeplab_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/SaliencyDetect/networks/deeplab_resnet.py -------------------------------------------------------------------------------- /datasets/generates/SaliencyDetect/networks/joint_poolnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/SaliencyDetect/networks/joint_poolnet.py -------------------------------------------------------------------------------- /datasets/generates/SaliencyDetect/networks/poolnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/SaliencyDetect/networks/poolnet.py -------------------------------------------------------------------------------- /datasets/generates/SaliencyDetect/networks/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/SaliencyDetect/networks/vgg.py -------------------------------------------------------------------------------- /datasets/generates/SaliencyDetect/saliency_detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/SaliencyDetect/saliency_detect.py -------------------------------------------------------------------------------- /datasets/generates/palette_mask_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/palette_mask_generate.py -------------------------------------------------------------------------------- /datasets/generates/saliency_mask_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/datasets/generates/saliency_mask_generate.py -------------------------------------------------------------------------------- /options/infer/infer_rsfnet_global_fivek.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/infer/infer_rsfnet_global_fivek.yml -------------------------------------------------------------------------------- /options/infer/infer_rsfnet_map_fivek.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/infer/infer_rsfnet_map_fivek.yml -------------------------------------------------------------------------------- /options/infer/infer_rsfnet_map_ppr10k_a.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/infer/infer_rsfnet_map_ppr10k_a.yml -------------------------------------------------------------------------------- /options/infer/infer_rsfnet_map_ppr10k_b.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/infer/infer_rsfnet_map_ppr10k_b.yml -------------------------------------------------------------------------------- /options/infer/infer_rsfnet_map_ppr10k_c.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/infer/infer_rsfnet_map_ppr10k_c.yml -------------------------------------------------------------------------------- /options/infer/infer_rsfnet_palette_fivek.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/infer/infer_rsfnet_palette_fivek.yml -------------------------------------------------------------------------------- /options/infer/infer_rsfnet_saliency_fivek.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/infer/infer_rsfnet_saliency_fivek.yml -------------------------------------------------------------------------------- /options/infer/infer_rsfnet_saliency_ppr10k.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/infer/infer_rsfnet_saliency_ppr10k.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_global_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_global_fivek_zas.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_global_fivek_zwc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_global_fivek_zwc.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_map_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_map_fivek_zas.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_map_fivek_zwc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_map_fivek_zwc.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_map_ppr10k_a.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_map_ppr10k_a.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_map_ppr10k_b.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_map_ppr10k_b.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_map_ppr10k_c.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_map_ppr10k_c.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_maskin_palette_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_maskin_palette_fivek_zas.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_maskin_saliency_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_maskin_saliency_fivek_zas.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_palette_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_palette_fivek_zas.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_palette_fivek_zwc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_palette_fivek_zwc.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_saliency_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_saliency_fivek_zas.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_saliency_fivek_zwc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_saliency_fivek_zwc.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_saliency_ppr10k_a.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_saliency_ppr10k_a.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_saliency_ppr10k_b.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_saliency_ppr10k_b.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_saliency_ppr10k_c.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_saliency_ppr10k_c.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_semantic_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_semantic_fivek_zas.yml -------------------------------------------------------------------------------- /options/test/RSFNet/test_rsfnet_semantic_fivek_zwc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/test/RSFNet/test_rsfnet_semantic_fivek_zwc.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_global_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_global_fivek_zas.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_global_fivek_zwc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_global_fivek_zwc.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_map_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_map_fivek_zas.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_map_fivek_zwc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_map_fivek_zwc.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_map_ppr10k_a.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_map_ppr10k_a.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_map_ppr10k_b.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_map_ppr10k_b.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_map_ppr10k_c.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_map_ppr10k_c.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_maskin_palette_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_maskin_palette_fivek_zas.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_maskin_saliency_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_maskin_saliency_fivek_zas.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_palette_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_palette_fivek_zas.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_palette_fivek_zwc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_palette_fivek_zwc.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_saliency_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_saliency_fivek_zas.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_saliency_fivek_zwc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_saliency_fivek_zwc.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_saliency_ppr10k_a.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_saliency_ppr10k_a.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_saliency_ppr10k_b.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_saliency_ppr10k_b.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_saliency_ppr10k_c.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_saliency_ppr10k_c.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_semantic_fivek_zas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_semantic_fivek_zas.yml -------------------------------------------------------------------------------- /options/train/RSFNet/train_rsfnet_semantic_fivek_zwc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/options/train/RSFNet/train_rsfnet_semantic_fivek_zwc.yml -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vicky0522/RSFNet/HEAD/setup.py --------------------------------------------------------------------------------