├── README.md ├── VERSION ├── archs ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── psrt_recurrent_arch.cpython-38.pyc │ └── psrt_sliding_arch.cpython-38.pyc ├── psrt_recurrent_arch.py └── psrt_sliding_arch.py ├── assets ├── PSRT-recurrent.png ├── PSRT-sliding.png ├── Results.png └── patch_warp.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 ├── inference_psrtrecurrent_reds.py └── inference_psrtrecurrent_vimeo90k.py ├── models ├── __init__.py └── recurrent_mixprecision_model.py ├── options ├── 4126_PSRTRecurrent_mix_precision_REDS_600K_N16.yml └── 5123_PSRTRecurrent_mix_precision_Vimeo_300K_N14.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.cfg ├── setup.py ├── test.py ├── train.py └── version.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/README.md -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 0.1.0 2 | -------------------------------------------------------------------------------- /archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/archs/__init__.py -------------------------------------------------------------------------------- /archs/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/archs/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /archs/__pycache__/psrt_recurrent_arch.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/archs/__pycache__/psrt_recurrent_arch.cpython-38.pyc -------------------------------------------------------------------------------- /archs/__pycache__/psrt_sliding_arch.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/archs/__pycache__/psrt_sliding_arch.cpython-38.pyc -------------------------------------------------------------------------------- /archs/psrt_recurrent_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/archs/psrt_recurrent_arch.py -------------------------------------------------------------------------------- /archs/psrt_sliding_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/archs/psrt_sliding_arch.py -------------------------------------------------------------------------------- /assets/PSRT-recurrent.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/assets/PSRT-recurrent.png -------------------------------------------------------------------------------- /assets/PSRT-sliding.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/assets/PSRT-sliding.png -------------------------------------------------------------------------------- /assets/Results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/assets/Results.png -------------------------------------------------------------------------------- /assets/patch_warp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/assets/patch_warp.png -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/meta_info_REDS_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/data/meta_info_REDS_GT.txt -------------------------------------------------------------------------------- /data/meta_info_Vimeo90K_train_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/data/meta_info_Vimeo90K_train_GT.txt -------------------------------------------------------------------------------- /dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/dist_train.sh -------------------------------------------------------------------------------- /experiments/pretrained_models/flownet/spynet_sintel_final-3d2a1287.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/experiments/pretrained_models/flownet/spynet_sintel_final-3d2a1287.pth -------------------------------------------------------------------------------- /inference/inference_psrtrecurrent_reds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/inference/inference_psrtrecurrent_reds.py -------------------------------------------------------------------------------- /inference/inference_psrtrecurrent_vimeo90k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/inference/inference_psrtrecurrent_vimeo90k.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/recurrent_mixprecision_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/models/recurrent_mixprecision_model.py -------------------------------------------------------------------------------- /options/4126_PSRTRecurrent_mix_precision_REDS_600K_N16.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/options/4126_PSRTRecurrent_mix_precision_REDS_600K_N16.yml -------------------------------------------------------------------------------- /options/5123_PSRTRecurrent_mix_precision_Vimeo_300K_N14.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/options/5123_PSRTRecurrent_mix_precision_Vimeo_300K_N14.yml -------------------------------------------------------------------------------- /recurrent_mix_precision_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/recurrent_mix_precision_train.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/compare_psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/scripts/compare_psnr.py -------------------------------------------------------------------------------- /scripts/downsample_512_vfhq_matlab_x4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/scripts/downsample_512_vfhq_matlab_x4.py -------------------------------------------------------------------------------- /scripts/downsample_center_crop_hdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/scripts/downsample_center_crop_hdr.py -------------------------------------------------------------------------------- /scripts/prepare_example_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/scripts/prepare_example_data.py -------------------------------------------------------------------------------- /scripts/rename.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/scripts/rename.py -------------------------------------------------------------------------------- /scripts/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/scripts/utils.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/setup.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/train.py -------------------------------------------------------------------------------- /version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPixelGroup/RethinkVSRAlignment/HEAD/version.py --------------------------------------------------------------------------------