├── LICENSE.txt ├── README.md ├── VERSION.txt ├── archs ├── __init__.py ├── functions.py ├── iia_sliding_arch.py ├── iia_vsr_arch.py ├── mia_sliding_arch.py └── mia_vsr_arch.py ├── assets ├── MIAVSR-arch.png ├── Masked-IIAB.png └── Results.png ├── data ├── __init__.py ├── meta_info_REDS_GT.txt └── meta_info_Vimeo90K_train_GT.txt ├── dist_train.sh ├── experiments └── pretrained_models │ └── flownet │ └── spynet_sintel_final-3d2a1287.pth ├── inference ├── __init__.py ├── inference_miavsr_reds.py └── inference_miavsr_vimeo90k.py ├── losses ├── __init__.py ├── loss_util.py └── sparsity_loss.py ├── models ├── __init__.py ├── recurrent_mixprecision_model.py ├── sr_model.py ├── video_base_model.py └── video_recurrent_model.py ├── options ├── 4126_MIAVSR_mix_precision_REDS_600K_N1.yml └── 5123_MIAVSR_mix_precision_Vimeo_300K_N10.yml ├── recurrent_mix_precision_train.py ├── requirements.txt ├── scripts ├── compare_psnr.py ├── downsample_512_vfhq_matlab_x4.py ├── downsample_center_crop_hdr.py ├── prepare_example_data.py ├── rename.py └── utils.py ├── setup.py ├── test.py ├── train.py └── version.py /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/README.md -------------------------------------------------------------------------------- /VERSION.txt: -------------------------------------------------------------------------------- 1 | 0.1.0 2 | -------------------------------------------------------------------------------- /archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/archs/__init__.py -------------------------------------------------------------------------------- /archs/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/archs/functions.py -------------------------------------------------------------------------------- /archs/iia_sliding_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/archs/iia_sliding_arch.py -------------------------------------------------------------------------------- /archs/iia_vsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/archs/iia_vsr_arch.py -------------------------------------------------------------------------------- /archs/mia_sliding_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/archs/mia_sliding_arch.py -------------------------------------------------------------------------------- /archs/mia_vsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/archs/mia_vsr_arch.py -------------------------------------------------------------------------------- /assets/MIAVSR-arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/assets/MIAVSR-arch.png -------------------------------------------------------------------------------- /assets/Masked-IIAB.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/assets/Masked-IIAB.png -------------------------------------------------------------------------------- /assets/Results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/assets/Results.png -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/meta_info_REDS_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/data/meta_info_REDS_GT.txt -------------------------------------------------------------------------------- /data/meta_info_Vimeo90K_train_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/data/meta_info_Vimeo90K_train_GT.txt -------------------------------------------------------------------------------- /dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/dist_train.sh -------------------------------------------------------------------------------- /experiments/pretrained_models/flownet/spynet_sintel_final-3d2a1287.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/experiments/pretrained_models/flownet/spynet_sintel_final-3d2a1287.pth -------------------------------------------------------------------------------- /inference/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /inference/inference_miavsr_reds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/inference/inference_miavsr_reds.py -------------------------------------------------------------------------------- /inference/inference_miavsr_vimeo90k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/inference/inference_miavsr_vimeo90k.py -------------------------------------------------------------------------------- /losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/losses/__init__.py -------------------------------------------------------------------------------- /losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/losses/loss_util.py -------------------------------------------------------------------------------- /losses/sparsity_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/losses/sparsity_loss.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/recurrent_mixprecision_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/models/recurrent_mixprecision_model.py -------------------------------------------------------------------------------- /models/sr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/models/sr_model.py -------------------------------------------------------------------------------- /models/video_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/models/video_base_model.py -------------------------------------------------------------------------------- /models/video_recurrent_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/models/video_recurrent_model.py -------------------------------------------------------------------------------- /options/4126_MIAVSR_mix_precision_REDS_600K_N1.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/options/4126_MIAVSR_mix_precision_REDS_600K_N1.yml -------------------------------------------------------------------------------- /options/5123_MIAVSR_mix_precision_Vimeo_300K_N10.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/options/5123_MIAVSR_mix_precision_Vimeo_300K_N10.yml -------------------------------------------------------------------------------- /recurrent_mix_precision_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/recurrent_mix_precision_train.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/compare_psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/scripts/compare_psnr.py -------------------------------------------------------------------------------- /scripts/downsample_512_vfhq_matlab_x4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/scripts/downsample_512_vfhq_matlab_x4.py -------------------------------------------------------------------------------- /scripts/downsample_center_crop_hdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/scripts/downsample_center_crop_hdr.py -------------------------------------------------------------------------------- /scripts/prepare_example_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/scripts/prepare_example_data.py -------------------------------------------------------------------------------- /scripts/rename.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/scripts/rename.py -------------------------------------------------------------------------------- /scripts/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/scripts/utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/setup.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/train.py -------------------------------------------------------------------------------- /version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LabShuHangGU/MIA-VSR/HEAD/version.py --------------------------------------------------------------------------------