├── .gitignore ├── LICENSE ├── README.md ├── assets ├── Picture1.png ├── Picture_H.png ├── interactive_images │ ├── a0002_1_5_input.jpg │ └── a0002_1_5_mask.png └── interactive_images_H │ ├── a0782_1_5_input.jpg │ └── a0782_1_5_mask.png ├── config.yml ├── config_test_HR.yml ├── config_test_LR.yml ├── config_test_adobe1024.yml ├── config_test_adobe2048.yml ├── iharm ├── data │ ├── base.py │ ├── compose.py │ ├── hdataset.py │ ├── hsv_utils.py │ └── transforms.py ├── engine │ ├── balanced_dp.py │ ├── event_loop.py │ ├── mhead_trainer.py │ ├── optimizer.py │ ├── simple_trainer.py │ ├── upsample_dccf_backbone_trainer.py │ ├── upsample_hsl_mc_trainer.py │ └── upsample_trainer.py ├── inference │ ├── evaluation.py │ ├── metrics.py │ ├── predictor.py │ ├── predictor_upsample_hsl.py │ ├── predictor_upsample_hsl_nobackbone.py │ ├── predictor_upsample_interactive.py │ ├── transforms.py │ └── utils.py ├── mconfigs │ ├── __init__.py │ ├── backboned.py │ └── base.py ├── model │ ├── backboned │ │ ├── __init__.py │ │ ├── deeplab.py │ │ ├── hrnet.py │ │ ├── hrnet_v2.py │ │ ├── ih_model.py │ │ └── ih_model_v2.py │ ├── base │ │ ├── __init__.py │ │ ├── dih_model.py │ │ ├── dih_model_v3.py │ │ ├── iseunet_v1.py │ │ └── ssam_model.py │ ├── gan_model.py │ ├── initializer.py │ ├── losses.py │ ├── metrics.py │ ├── modeling │ │ ├── ViT_configs.py │ │ ├── basic_blocks.py │ │ ├── conv_autoencoder.py │ │ ├── deeplab_v3.py │ │ ├── hrnet_ocr.py │ │ ├── ocr.py │ │ ├── resnet.py │ │ ├── resnetv1b.py │ │ ├── transUnet.py │ │ └── unet.py │ ├── modifiers.py │ ├── ops.py │ ├── ps_filters.py │ └── syncbn │ │ ├── LICENSE │ │ ├── README.md │ │ ├── __init__.py │ │ └── modules │ │ ├── __init__.py │ │ ├── functional │ │ ├── __init__.py │ │ ├── _csrc.py │ │ ├── csrc │ │ │ ├── bn.h │ │ │ ├── cuda │ │ │ │ ├── bn_cuda.cu │ │ │ │ ├── common.h │ │ │ │ └── ext_lib.h │ │ │ └── ext_lib.cpp │ │ └── syncbn.py │ │ └── nn │ │ ├── __init__.py │ │ └── syncbn.py └── utils │ ├── exp.py │ ├── log.py │ └── misc.py ├── models └── fixed256 │ ├── hrnet18s_v2p_idih.py │ ├── hrnet18s_v2p_idih_upsample_hsl_refine_HR.py │ ├── hrnet18s_v2p_idih_upsample_hsl_refine_HR_sat_clamp.py │ ├── hrnet18s_v2p_idih_upsample_hsl_refine_LR.py │ ├── hrnet18s_v2p_idih_upsample_hsl_refine_LR_sat_clamp.py │ ├── idih_dccf_HR_hsv1_clamp.py │ ├── idih_dccf_LR_hsv1_clamp.py │ ├── issam_dccf_HR_adobe1024.py │ ├── issam_dccf_HR_adobe2048.py │ ├── issam_dccf_HR_hsv_1_clamp.py │ └── issam_dccf_LR_hsv_1_clamp.py ├── notebooks ├── resize_dataset.py ├── resize_dataset_1024_1024.py └── resize_dataset_2048_2048.py ├── requirements.txt ├── runs ├── interactive_experiments.sh ├── test_baselines_idih.sh ├── test_baselines_idih_hrnet18s_v2p.sh ├── test_baselines_issam.sh ├── test_dccf_idih_hrnet18s_v2p_pretrain.sh ├── test_dccf_idih_pretrain.sh ├── test_dccf_issam_pretrain.sh ├── train_dccf_idih.sh ├── train_dccf_idih_hrnet18s_v2p.sh ├── train_dccf_issam.sh └── train_dccf_issam_finetune_adobe.sh ├── scripts ├── evaluate_model_rawsize.py ├── evaluate_upsample_refiner.py └── predict_interactive.py └── train.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | harmonization_exps/ 3 | pretrained_models/ 4 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/README.md -------------------------------------------------------------------------------- /assets/Picture1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/assets/Picture1.png -------------------------------------------------------------------------------- /assets/Picture_H.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/assets/Picture_H.png -------------------------------------------------------------------------------- /assets/interactive_images/a0002_1_5_input.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/assets/interactive_images/a0002_1_5_input.jpg -------------------------------------------------------------------------------- /assets/interactive_images/a0002_1_5_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/assets/interactive_images/a0002_1_5_mask.png -------------------------------------------------------------------------------- /assets/interactive_images_H/a0782_1_5_input.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/assets/interactive_images_H/a0782_1_5_input.jpg -------------------------------------------------------------------------------- /assets/interactive_images_H/a0782_1_5_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/assets/interactive_images_H/a0782_1_5_mask.png -------------------------------------------------------------------------------- /config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/config.yml -------------------------------------------------------------------------------- /config_test_HR.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/config_test_HR.yml -------------------------------------------------------------------------------- /config_test_LR.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/config_test_LR.yml -------------------------------------------------------------------------------- /config_test_adobe1024.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/config_test_adobe1024.yml -------------------------------------------------------------------------------- /config_test_adobe2048.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/config_test_adobe2048.yml -------------------------------------------------------------------------------- /iharm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/data/base.py -------------------------------------------------------------------------------- /iharm/data/compose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/data/compose.py -------------------------------------------------------------------------------- /iharm/data/hdataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/data/hdataset.py -------------------------------------------------------------------------------- /iharm/data/hsv_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/data/hsv_utils.py -------------------------------------------------------------------------------- /iharm/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/data/transforms.py -------------------------------------------------------------------------------- /iharm/engine/balanced_dp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/engine/balanced_dp.py -------------------------------------------------------------------------------- /iharm/engine/event_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/engine/event_loop.py -------------------------------------------------------------------------------- /iharm/engine/mhead_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/engine/mhead_trainer.py -------------------------------------------------------------------------------- /iharm/engine/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/engine/optimizer.py -------------------------------------------------------------------------------- /iharm/engine/simple_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/engine/simple_trainer.py -------------------------------------------------------------------------------- /iharm/engine/upsample_dccf_backbone_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/engine/upsample_dccf_backbone_trainer.py -------------------------------------------------------------------------------- /iharm/engine/upsample_hsl_mc_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/engine/upsample_hsl_mc_trainer.py -------------------------------------------------------------------------------- /iharm/engine/upsample_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/engine/upsample_trainer.py -------------------------------------------------------------------------------- /iharm/inference/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/inference/evaluation.py -------------------------------------------------------------------------------- /iharm/inference/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/inference/metrics.py -------------------------------------------------------------------------------- /iharm/inference/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/inference/predictor.py -------------------------------------------------------------------------------- /iharm/inference/predictor_upsample_hsl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/inference/predictor_upsample_hsl.py -------------------------------------------------------------------------------- /iharm/inference/predictor_upsample_hsl_nobackbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/inference/predictor_upsample_hsl_nobackbone.py -------------------------------------------------------------------------------- /iharm/inference/predictor_upsample_interactive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/inference/predictor_upsample_interactive.py -------------------------------------------------------------------------------- /iharm/inference/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/inference/transforms.py -------------------------------------------------------------------------------- /iharm/inference/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/inference/utils.py -------------------------------------------------------------------------------- /iharm/mconfigs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/mconfigs/__init__.py -------------------------------------------------------------------------------- /iharm/mconfigs/backboned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/mconfigs/backboned.py -------------------------------------------------------------------------------- /iharm/mconfigs/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/mconfigs/base.py -------------------------------------------------------------------------------- /iharm/model/backboned/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/backboned/__init__.py -------------------------------------------------------------------------------- /iharm/model/backboned/deeplab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/backboned/deeplab.py -------------------------------------------------------------------------------- /iharm/model/backboned/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/backboned/hrnet.py -------------------------------------------------------------------------------- /iharm/model/backboned/hrnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/backboned/hrnet_v2.py -------------------------------------------------------------------------------- /iharm/model/backboned/ih_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/backboned/ih_model.py -------------------------------------------------------------------------------- /iharm/model/backboned/ih_model_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/backboned/ih_model_v2.py -------------------------------------------------------------------------------- /iharm/model/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/base/__init__.py -------------------------------------------------------------------------------- /iharm/model/base/dih_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/base/dih_model.py -------------------------------------------------------------------------------- /iharm/model/base/dih_model_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/base/dih_model_v3.py -------------------------------------------------------------------------------- /iharm/model/base/iseunet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/base/iseunet_v1.py -------------------------------------------------------------------------------- /iharm/model/base/ssam_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/base/ssam_model.py -------------------------------------------------------------------------------- /iharm/model/gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/gan_model.py -------------------------------------------------------------------------------- /iharm/model/initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/initializer.py -------------------------------------------------------------------------------- /iharm/model/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/losses.py -------------------------------------------------------------------------------- /iharm/model/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/metrics.py -------------------------------------------------------------------------------- /iharm/model/modeling/ViT_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/modeling/ViT_configs.py -------------------------------------------------------------------------------- /iharm/model/modeling/basic_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/modeling/basic_blocks.py -------------------------------------------------------------------------------- /iharm/model/modeling/conv_autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/modeling/conv_autoencoder.py -------------------------------------------------------------------------------- /iharm/model/modeling/deeplab_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/modeling/deeplab_v3.py -------------------------------------------------------------------------------- /iharm/model/modeling/hrnet_ocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/modeling/hrnet_ocr.py -------------------------------------------------------------------------------- /iharm/model/modeling/ocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/modeling/ocr.py -------------------------------------------------------------------------------- /iharm/model/modeling/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/modeling/resnet.py -------------------------------------------------------------------------------- /iharm/model/modeling/resnetv1b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/modeling/resnetv1b.py -------------------------------------------------------------------------------- /iharm/model/modeling/transUnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/modeling/transUnet.py -------------------------------------------------------------------------------- /iharm/model/modeling/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/modeling/unet.py -------------------------------------------------------------------------------- /iharm/model/modifiers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/modifiers.py -------------------------------------------------------------------------------- /iharm/model/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/ops.py -------------------------------------------------------------------------------- /iharm/model/ps_filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/ps_filters.py -------------------------------------------------------------------------------- /iharm/model/syncbn/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/syncbn/LICENSE -------------------------------------------------------------------------------- /iharm/model/syncbn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/syncbn/README.md -------------------------------------------------------------------------------- /iharm/model/syncbn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/__init__.py: -------------------------------------------------------------------------------- 1 | from .syncbn import batchnorm2d_sync 2 | -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/_csrc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/syncbn/modules/functional/_csrc.py -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/csrc/bn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/syncbn/modules/functional/csrc/bn.h -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/csrc/cuda/bn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/syncbn/modules/functional/csrc/cuda/bn_cuda.cu -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/csrc/cuda/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/syncbn/modules/functional/csrc/cuda/common.h -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/csrc/cuda/ext_lib.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/syncbn/modules/functional/csrc/cuda/ext_lib.h -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/csrc/ext_lib.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/syncbn/modules/functional/csrc/ext_lib.cpp -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/syncbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/syncbn/modules/functional/syncbn.py -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/nn/__init__.py: -------------------------------------------------------------------------------- 1 | from .syncbn import * 2 | -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/nn/syncbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/model/syncbn/modules/nn/syncbn.py -------------------------------------------------------------------------------- /iharm/utils/exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/utils/exp.py -------------------------------------------------------------------------------- /iharm/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/utils/log.py -------------------------------------------------------------------------------- /iharm/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/iharm/utils/misc.py -------------------------------------------------------------------------------- /models/fixed256/hrnet18s_v2p_idih.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/models/fixed256/hrnet18s_v2p_idih.py -------------------------------------------------------------------------------- /models/fixed256/hrnet18s_v2p_idih_upsample_hsl_refine_HR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/models/fixed256/hrnet18s_v2p_idih_upsample_hsl_refine_HR.py -------------------------------------------------------------------------------- /models/fixed256/hrnet18s_v2p_idih_upsample_hsl_refine_HR_sat_clamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/models/fixed256/hrnet18s_v2p_idih_upsample_hsl_refine_HR_sat_clamp.py -------------------------------------------------------------------------------- /models/fixed256/hrnet18s_v2p_idih_upsample_hsl_refine_LR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/models/fixed256/hrnet18s_v2p_idih_upsample_hsl_refine_LR.py -------------------------------------------------------------------------------- /models/fixed256/hrnet18s_v2p_idih_upsample_hsl_refine_LR_sat_clamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/models/fixed256/hrnet18s_v2p_idih_upsample_hsl_refine_LR_sat_clamp.py -------------------------------------------------------------------------------- /models/fixed256/idih_dccf_HR_hsv1_clamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/models/fixed256/idih_dccf_HR_hsv1_clamp.py -------------------------------------------------------------------------------- /models/fixed256/idih_dccf_LR_hsv1_clamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/models/fixed256/idih_dccf_LR_hsv1_clamp.py -------------------------------------------------------------------------------- /models/fixed256/issam_dccf_HR_adobe1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/models/fixed256/issam_dccf_HR_adobe1024.py -------------------------------------------------------------------------------- /models/fixed256/issam_dccf_HR_adobe2048.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/models/fixed256/issam_dccf_HR_adobe2048.py -------------------------------------------------------------------------------- /models/fixed256/issam_dccf_HR_hsv_1_clamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/models/fixed256/issam_dccf_HR_hsv_1_clamp.py -------------------------------------------------------------------------------- /models/fixed256/issam_dccf_LR_hsv_1_clamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/models/fixed256/issam_dccf_LR_hsv_1_clamp.py -------------------------------------------------------------------------------- /notebooks/resize_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/notebooks/resize_dataset.py -------------------------------------------------------------------------------- /notebooks/resize_dataset_1024_1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/notebooks/resize_dataset_1024_1024.py -------------------------------------------------------------------------------- /notebooks/resize_dataset_2048_2048.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/notebooks/resize_dataset_2048_2048.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/requirements.txt -------------------------------------------------------------------------------- /runs/interactive_experiments.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/runs/interactive_experiments.sh -------------------------------------------------------------------------------- /runs/test_baselines_idih.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/runs/test_baselines_idih.sh -------------------------------------------------------------------------------- /runs/test_baselines_idih_hrnet18s_v2p.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/runs/test_baselines_idih_hrnet18s_v2p.sh -------------------------------------------------------------------------------- /runs/test_baselines_issam.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/runs/test_baselines_issam.sh -------------------------------------------------------------------------------- /runs/test_dccf_idih_hrnet18s_v2p_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/runs/test_dccf_idih_hrnet18s_v2p_pretrain.sh -------------------------------------------------------------------------------- /runs/test_dccf_idih_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/runs/test_dccf_idih_pretrain.sh -------------------------------------------------------------------------------- /runs/test_dccf_issam_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/runs/test_dccf_issam_pretrain.sh -------------------------------------------------------------------------------- /runs/train_dccf_idih.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/runs/train_dccf_idih.sh -------------------------------------------------------------------------------- /runs/train_dccf_idih_hrnet18s_v2p.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/runs/train_dccf_idih_hrnet18s_v2p.sh -------------------------------------------------------------------------------- /runs/train_dccf_issam.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/runs/train_dccf_issam.sh -------------------------------------------------------------------------------- /runs/train_dccf_issam_finetune_adobe.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/runs/train_dccf_issam_finetune_adobe.sh -------------------------------------------------------------------------------- /scripts/evaluate_model_rawsize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/scripts/evaluate_model_rawsize.py -------------------------------------------------------------------------------- /scripts/evaluate_upsample_refiner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/scripts/evaluate_upsample_refiner.py -------------------------------------------------------------------------------- /scripts/predict_interactive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/scripts/predict_interactive.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rockeyben/DCCF/HEAD/train.py --------------------------------------------------------------------------------