├── .gitignore ├── LICENSE ├── README.md ├── checkpoints └── README.md ├── configs ├── TMNet_multiple_frames.yaml ├── TMNet_single_frame.yaml └── debug.yaml ├── data ├── README.md ├── Vimeo7_dataset.py ├── __init__.py ├── create_lmdb_list_adobe240fps.py ├── create_lmdb_mp_Vimeo90K_HR.py ├── create_lmdb_mp_Vimeo90K_LR.py ├── create_lmdb_mp_adobe240fps_HR.py ├── create_lmdb_mp_adobe240fps_LR.py ├── data_sampler.py ├── data_scripts_check │ ├── zsm_old │ │ ├── create_lmdb_mp.py │ │ ├── generate_LR_Vimeo90K.py │ │ ├── generate_mod_LR_bic.py │ │ └── sep_vimeo_list.py │ └── zsm_time_fine-tune │ │ ├── create_lmdb_mp.py │ │ ├── generate_LR_Vimeo90K.py │ │ ├── generate_mod_LR_bic.py │ │ └── sep_vimeo_list.py ├── data_util.py ├── generate_LR_Vimeo90K.py ├── generate_data_adobe240fps.py ├── generate_frames_from_adobe240fps.py ├── generate_mod_LR_bic.py ├── get_adobe240fps_pkl.sh ├── sep_vimeo_list_Vimeo90K.py ├── util.py └── util_old.py ├── datasets └── README.md ├── evaluations └── README.md ├── experiments └── README.md ├── images ├── cfi.png ├── logo_0.png ├── logo_1.png ├── method.png ├── quantitative_results.png ├── visual_results_0.png ├── visual_results_1.png └── visual_results_2.png ├── models ├── VideoSR_base_model.py ├── __init__.py ├── base_model.py ├── lr_scheduler.py ├── modules │ ├── DCNv2 │ │ ├── .gitignore │ │ ├── LICENSE │ │ ├── README.md │ │ ├── __init__.py │ │ ├── dcn_v2.py │ │ ├── make.sh │ │ ├── setup.py │ │ ├── src │ │ │ ├── cpu │ │ │ │ ├── dcn_v2_cpu.cpp │ │ │ │ └── vision.h │ │ │ ├── cuda │ │ │ │ ├── dcn_v2_cuda.cu │ │ │ │ ├── dcn_v2_im2col_cuda.cu │ │ │ │ ├── dcn_v2_im2col_cuda.h │ │ │ │ ├── dcn_v2_psroi_pooling_cuda.cu │ │ │ │ └── vision.h │ │ │ ├── dcn_v2.h │ │ │ └── vision.cpp │ │ └── test.py │ ├── STVSR.py │ ├── __init__.py │ ├── convlstm.py │ ├── loss.py │ └── module_util.py └── networks.py ├── options ├── __init__.py └── options.py ├── test_multiple_frames.py ├── test_single_frame.py ├── train.py └── utils ├── __init__.py ├── evaluation_multiple_frames.py ├── evaluation_single_frame.py ├── make_video.py └── util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/README.md -------------------------------------------------------------------------------- /checkpoints/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/checkpoints/README.md -------------------------------------------------------------------------------- /configs/TMNet_multiple_frames.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/configs/TMNet_multiple_frames.yaml -------------------------------------------------------------------------------- /configs/TMNet_single_frame.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/configs/TMNet_single_frame.yaml -------------------------------------------------------------------------------- /configs/debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/configs/debug.yaml -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/README.md -------------------------------------------------------------------------------- /data/Vimeo7_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/Vimeo7_dataset.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/create_lmdb_list_adobe240fps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/create_lmdb_list_adobe240fps.py -------------------------------------------------------------------------------- /data/create_lmdb_mp_Vimeo90K_HR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/create_lmdb_mp_Vimeo90K_HR.py -------------------------------------------------------------------------------- /data/create_lmdb_mp_Vimeo90K_LR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/create_lmdb_mp_Vimeo90K_LR.py -------------------------------------------------------------------------------- /data/create_lmdb_mp_adobe240fps_HR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/create_lmdb_mp_adobe240fps_HR.py -------------------------------------------------------------------------------- /data/create_lmdb_mp_adobe240fps_LR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/create_lmdb_mp_adobe240fps_LR.py -------------------------------------------------------------------------------- /data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/data_sampler.py -------------------------------------------------------------------------------- /data/data_scripts_check/zsm_old/create_lmdb_mp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/data_scripts_check/zsm_old/create_lmdb_mp.py -------------------------------------------------------------------------------- /data/data_scripts_check/zsm_old/generate_LR_Vimeo90K.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/data_scripts_check/zsm_old/generate_LR_Vimeo90K.py -------------------------------------------------------------------------------- /data/data_scripts_check/zsm_old/generate_mod_LR_bic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/data_scripts_check/zsm_old/generate_mod_LR_bic.py -------------------------------------------------------------------------------- /data/data_scripts_check/zsm_old/sep_vimeo_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/data_scripts_check/zsm_old/sep_vimeo_list.py -------------------------------------------------------------------------------- /data/data_scripts_check/zsm_time_fine-tune/create_lmdb_mp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/data_scripts_check/zsm_time_fine-tune/create_lmdb_mp.py -------------------------------------------------------------------------------- /data/data_scripts_check/zsm_time_fine-tune/generate_LR_Vimeo90K.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/data_scripts_check/zsm_time_fine-tune/generate_LR_Vimeo90K.py -------------------------------------------------------------------------------- /data/data_scripts_check/zsm_time_fine-tune/generate_mod_LR_bic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/data_scripts_check/zsm_time_fine-tune/generate_mod_LR_bic.py -------------------------------------------------------------------------------- /data/data_scripts_check/zsm_time_fine-tune/sep_vimeo_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/data_scripts_check/zsm_time_fine-tune/sep_vimeo_list.py -------------------------------------------------------------------------------- /data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/data_util.py -------------------------------------------------------------------------------- /data/generate_LR_Vimeo90K.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/generate_LR_Vimeo90K.py -------------------------------------------------------------------------------- /data/generate_data_adobe240fps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/generate_data_adobe240fps.py -------------------------------------------------------------------------------- /data/generate_frames_from_adobe240fps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/generate_frames_from_adobe240fps.py -------------------------------------------------------------------------------- /data/generate_mod_LR_bic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/generate_mod_LR_bic.py -------------------------------------------------------------------------------- /data/get_adobe240fps_pkl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/get_adobe240fps_pkl.sh -------------------------------------------------------------------------------- /data/sep_vimeo_list_Vimeo90K.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/sep_vimeo_list_Vimeo90K.py -------------------------------------------------------------------------------- /data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/util.py -------------------------------------------------------------------------------- /data/util_old.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/data/util_old.py -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- 1 | You can put the dataset(or soft link) into this folder -------------------------------------------------------------------------------- /evaluations/README.md: -------------------------------------------------------------------------------- 1 | The evaluation results are placed into this folder. -------------------------------------------------------------------------------- /experiments/README.md: -------------------------------------------------------------------------------- 1 | The training files are place into this folder. -------------------------------------------------------------------------------- /images/cfi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/images/cfi.png -------------------------------------------------------------------------------- /images/logo_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/images/logo_0.png -------------------------------------------------------------------------------- /images/logo_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/images/logo_1.png -------------------------------------------------------------------------------- /images/method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/images/method.png -------------------------------------------------------------------------------- /images/quantitative_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/images/quantitative_results.png -------------------------------------------------------------------------------- /images/visual_results_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/images/visual_results_0.png -------------------------------------------------------------------------------- /images/visual_results_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/images/visual_results_1.png -------------------------------------------------------------------------------- /images/visual_results_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/images/visual_results_2.png -------------------------------------------------------------------------------- /models/VideoSR_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/VideoSR_base_model.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/base_model.py -------------------------------------------------------------------------------- /models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/lr_scheduler.py -------------------------------------------------------------------------------- /models/modules/DCNv2/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/.gitignore -------------------------------------------------------------------------------- /models/modules/DCNv2/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/LICENSE -------------------------------------------------------------------------------- /models/modules/DCNv2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/README.md -------------------------------------------------------------------------------- /models/modules/DCNv2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/modules/DCNv2/dcn_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/dcn_v2.py -------------------------------------------------------------------------------- /models/modules/DCNv2/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/make.sh -------------------------------------------------------------------------------- /models/modules/DCNv2/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/setup.py -------------------------------------------------------------------------------- /models/modules/DCNv2/src/cpu/dcn_v2_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/src/cpu/dcn_v2_cpu.cpp -------------------------------------------------------------------------------- /models/modules/DCNv2/src/cpu/vision.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/src/cpu/vision.h -------------------------------------------------------------------------------- /models/modules/DCNv2/src/cuda/dcn_v2_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/src/cuda/dcn_v2_cuda.cu -------------------------------------------------------------------------------- /models/modules/DCNv2/src/cuda/dcn_v2_im2col_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/src/cuda/dcn_v2_im2col_cuda.cu -------------------------------------------------------------------------------- /models/modules/DCNv2/src/cuda/dcn_v2_im2col_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/src/cuda/dcn_v2_im2col_cuda.h -------------------------------------------------------------------------------- /models/modules/DCNv2/src/cuda/dcn_v2_psroi_pooling_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/src/cuda/dcn_v2_psroi_pooling_cuda.cu -------------------------------------------------------------------------------- /models/modules/DCNv2/src/cuda/vision.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/src/cuda/vision.h -------------------------------------------------------------------------------- /models/modules/DCNv2/src/dcn_v2.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/src/dcn_v2.h -------------------------------------------------------------------------------- /models/modules/DCNv2/src/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/src/vision.cpp -------------------------------------------------------------------------------- /models/modules/DCNv2/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/DCNv2/test.py -------------------------------------------------------------------------------- /models/modules/STVSR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/STVSR.py -------------------------------------------------------------------------------- /models/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/modules/convlstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/convlstm.py -------------------------------------------------------------------------------- /models/modules/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/loss.py -------------------------------------------------------------------------------- /models/modules/module_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/modules/module_util.py -------------------------------------------------------------------------------- /models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/models/networks.py -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /options/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/options/options.py -------------------------------------------------------------------------------- /test_multiple_frames.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/test_multiple_frames.py -------------------------------------------------------------------------------- /test_single_frame.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/test_single_frame.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/evaluation_multiple_frames.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/utils/evaluation_multiple_frames.py -------------------------------------------------------------------------------- /utils/evaluation_single_frame.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/utils/evaluation_single_frame.py -------------------------------------------------------------------------------- /utils/make_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/utils/make_video.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CS-GangXu/TMNet/HEAD/utils/util.py --------------------------------------------------------------------------------