├── LICENSE ├── README.md ├── codes ├── data │ ├── GTLQ_dataset.py │ ├── GTLQnpy_dataset.py │ ├── GTLQx_dataset.py │ ├── GT_dataset.py │ ├── LQ_dataset.py │ ├── LRHR_PKL_dataset.py │ ├── __init__.py │ ├── data_sampler.py │ └── util.py ├── models │ ├── HCFlow_Rescaling_model.py │ ├── HCFlow_SR_model.py │ ├── __init__.py │ ├── base_model.py │ ├── lr_scheduler.py │ ├── modules │ │ ├── ActNorms.py │ │ ├── AffineCouplings.py │ │ ├── Basic.py │ │ ├── ConditionalFlow.py │ │ ├── FlowNet_Rescaling_x4.py │ │ ├── FlowNet_SR_x4.py │ │ ├── FlowNet_SR_x8.py │ │ ├── FlowStep.py │ │ ├── HCFlowNet_Rescaling_arch.py │ │ ├── HCFlowNet_SR_arch.py │ │ ├── Permutations.py │ │ ├── discriminator_vgg_arch.py │ │ ├── loss.py │ │ ├── module_util.py │ │ └── thops.py │ └── networks.py ├── options │ ├── options.py │ ├── test │ │ ├── test_Rescaling_DF2K_4X_HCFlow.yml │ │ ├── test_SR_CelebA_8X_HCFlow.yml │ │ └── test_SR_DF2K_4X_HCFlow.yml │ └── train │ │ ├── train_Rescaling_DF2K_4X_HCFlow.yml │ │ ├── train_SR_CelebA_8X_HCFlow++.yml │ │ ├── train_SR_CelebA_8X_HCFlow+.yml │ │ ├── train_SR_CelebA_8X_HCFlow.yml │ │ ├── train_SR_DF2K_4X_HCFlow++.yml │ │ ├── train_SR_DF2K_4X_HCFlow+.yml │ │ └── train_SR_DF2K_4X_HCFlow.yml ├── scripts │ ├── png2npy.py │ └── prepare_data_pkl.py ├── test_HCFlow.py ├── train_HCFlow.py └── utils │ ├── dist_util.py │ ├── imresize.py │ ├── misc.py │ ├── timer.py │ └── util.py ├── cog.yaml ├── datasets ├── example_face_8X │ ├── HR │ │ ├── 110.png │ │ ├── 16.png │ │ ├── 217.png │ │ ├── 253.png │ │ ├── 334.png │ │ └── 36.png │ └── LR │ │ ├── 110.png │ │ ├── 16.png │ │ ├── 217.png │ │ ├── 253.png │ │ ├── 334.png │ │ └── 36.png └── example_general_4X │ ├── HR │ └── butterfly.png │ └── LR │ └── butterfly.png ├── download-weights ├── experiments └── pretrained_models │ └── README.md ├── illustrations ├── architecture.png ├── computation_graph.png └── face_result.png ├── predict.py └── requirements.txt /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/README.md -------------------------------------------------------------------------------- /codes/data/GTLQ_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/data/GTLQ_dataset.py -------------------------------------------------------------------------------- /codes/data/GTLQnpy_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/data/GTLQnpy_dataset.py -------------------------------------------------------------------------------- /codes/data/GTLQx_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/data/GTLQx_dataset.py -------------------------------------------------------------------------------- /codes/data/GT_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/data/GT_dataset.py -------------------------------------------------------------------------------- /codes/data/LQ_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/data/LQ_dataset.py -------------------------------------------------------------------------------- /codes/data/LRHR_PKL_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/data/LRHR_PKL_dataset.py -------------------------------------------------------------------------------- /codes/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/data/__init__.py -------------------------------------------------------------------------------- /codes/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/data/data_sampler.py -------------------------------------------------------------------------------- /codes/data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/data/util.py -------------------------------------------------------------------------------- /codes/models/HCFlow_Rescaling_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/HCFlow_Rescaling_model.py -------------------------------------------------------------------------------- /codes/models/HCFlow_SR_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/HCFlow_SR_model.py -------------------------------------------------------------------------------- /codes/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/__init__.py -------------------------------------------------------------------------------- /codes/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/base_model.py -------------------------------------------------------------------------------- /codes/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/lr_scheduler.py -------------------------------------------------------------------------------- /codes/models/modules/ActNorms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/ActNorms.py -------------------------------------------------------------------------------- /codes/models/modules/AffineCouplings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/AffineCouplings.py -------------------------------------------------------------------------------- /codes/models/modules/Basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/Basic.py -------------------------------------------------------------------------------- /codes/models/modules/ConditionalFlow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/ConditionalFlow.py -------------------------------------------------------------------------------- /codes/models/modules/FlowNet_Rescaling_x4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/FlowNet_Rescaling_x4.py -------------------------------------------------------------------------------- /codes/models/modules/FlowNet_SR_x4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/FlowNet_SR_x4.py -------------------------------------------------------------------------------- /codes/models/modules/FlowNet_SR_x8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/FlowNet_SR_x8.py -------------------------------------------------------------------------------- /codes/models/modules/FlowStep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/FlowStep.py -------------------------------------------------------------------------------- /codes/models/modules/HCFlowNet_Rescaling_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/HCFlowNet_Rescaling_arch.py -------------------------------------------------------------------------------- /codes/models/modules/HCFlowNet_SR_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/HCFlowNet_SR_arch.py -------------------------------------------------------------------------------- /codes/models/modules/Permutations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/Permutations.py -------------------------------------------------------------------------------- /codes/models/modules/discriminator_vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/discriminator_vgg_arch.py -------------------------------------------------------------------------------- /codes/models/modules/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/loss.py -------------------------------------------------------------------------------- /codes/models/modules/module_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/module_util.py -------------------------------------------------------------------------------- /codes/models/modules/thops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/modules/thops.py -------------------------------------------------------------------------------- /codes/models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/models/networks.py -------------------------------------------------------------------------------- /codes/options/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/options/options.py -------------------------------------------------------------------------------- /codes/options/test/test_Rescaling_DF2K_4X_HCFlow.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/options/test/test_Rescaling_DF2K_4X_HCFlow.yml -------------------------------------------------------------------------------- /codes/options/test/test_SR_CelebA_8X_HCFlow.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/options/test/test_SR_CelebA_8X_HCFlow.yml -------------------------------------------------------------------------------- /codes/options/test/test_SR_DF2K_4X_HCFlow.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/options/test/test_SR_DF2K_4X_HCFlow.yml -------------------------------------------------------------------------------- /codes/options/train/train_Rescaling_DF2K_4X_HCFlow.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/options/train/train_Rescaling_DF2K_4X_HCFlow.yml -------------------------------------------------------------------------------- /codes/options/train/train_SR_CelebA_8X_HCFlow++.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/options/train/train_SR_CelebA_8X_HCFlow++.yml -------------------------------------------------------------------------------- /codes/options/train/train_SR_CelebA_8X_HCFlow+.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/options/train/train_SR_CelebA_8X_HCFlow+.yml -------------------------------------------------------------------------------- /codes/options/train/train_SR_CelebA_8X_HCFlow.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/options/train/train_SR_CelebA_8X_HCFlow.yml -------------------------------------------------------------------------------- /codes/options/train/train_SR_DF2K_4X_HCFlow++.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/options/train/train_SR_DF2K_4X_HCFlow++.yml -------------------------------------------------------------------------------- /codes/options/train/train_SR_DF2K_4X_HCFlow+.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/options/train/train_SR_DF2K_4X_HCFlow+.yml -------------------------------------------------------------------------------- /codes/options/train/train_SR_DF2K_4X_HCFlow.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/options/train/train_SR_DF2K_4X_HCFlow.yml -------------------------------------------------------------------------------- /codes/scripts/png2npy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/scripts/png2npy.py -------------------------------------------------------------------------------- /codes/scripts/prepare_data_pkl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/scripts/prepare_data_pkl.py -------------------------------------------------------------------------------- /codes/test_HCFlow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/test_HCFlow.py -------------------------------------------------------------------------------- /codes/train_HCFlow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/train_HCFlow.py -------------------------------------------------------------------------------- /codes/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/utils/dist_util.py -------------------------------------------------------------------------------- /codes/utils/imresize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/utils/imresize.py -------------------------------------------------------------------------------- /codes/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/utils/misc.py -------------------------------------------------------------------------------- /codes/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/utils/timer.py -------------------------------------------------------------------------------- /codes/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/codes/utils/util.py -------------------------------------------------------------------------------- /cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/cog.yaml -------------------------------------------------------------------------------- /datasets/example_face_8X/HR/110.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_face_8X/HR/110.png -------------------------------------------------------------------------------- /datasets/example_face_8X/HR/16.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_face_8X/HR/16.png -------------------------------------------------------------------------------- /datasets/example_face_8X/HR/217.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_face_8X/HR/217.png -------------------------------------------------------------------------------- /datasets/example_face_8X/HR/253.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_face_8X/HR/253.png -------------------------------------------------------------------------------- /datasets/example_face_8X/HR/334.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_face_8X/HR/334.png -------------------------------------------------------------------------------- /datasets/example_face_8X/HR/36.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_face_8X/HR/36.png -------------------------------------------------------------------------------- /datasets/example_face_8X/LR/110.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_face_8X/LR/110.png -------------------------------------------------------------------------------- /datasets/example_face_8X/LR/16.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_face_8X/LR/16.png -------------------------------------------------------------------------------- /datasets/example_face_8X/LR/217.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_face_8X/LR/217.png -------------------------------------------------------------------------------- /datasets/example_face_8X/LR/253.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_face_8X/LR/253.png -------------------------------------------------------------------------------- /datasets/example_face_8X/LR/334.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_face_8X/LR/334.png -------------------------------------------------------------------------------- /datasets/example_face_8X/LR/36.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_face_8X/LR/36.png -------------------------------------------------------------------------------- /datasets/example_general_4X/HR/butterfly.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_general_4X/HR/butterfly.png -------------------------------------------------------------------------------- /datasets/example_general_4X/LR/butterfly.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/datasets/example_general_4X/LR/butterfly.png -------------------------------------------------------------------------------- /download-weights: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/download-weights -------------------------------------------------------------------------------- /experiments/pretrained_models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/experiments/pretrained_models/README.md -------------------------------------------------------------------------------- /illustrations/architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/illustrations/architecture.png -------------------------------------------------------------------------------- /illustrations/computation_graph.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/illustrations/computation_graph.png -------------------------------------------------------------------------------- /illustrations/face_result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/illustrations/face_result.png -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/predict.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JingyunLiang/HCFlow/HEAD/requirements.txt --------------------------------------------------------------------------------