search

JingyunLiang / RVRT

Recurrent Video Restoration Transformer with Guided Deformable Attention (NeurlPS2022, official repository)
https://arxiv.org/abs/2206.02146
Other
354 stars 33 forks source link

How to download Vimeo-90K Test Datasets? #20

Open CuteYuzu opened 1 year ago

CuteYuzu commented 1 year ago

I run for this task: "python main_test_rvrt.py --task 002_RVRT_videosr_bi_Vimeo_14frames --folder_lq testsets/vimeo90k/vimeo_septuplet_matlabLRx4/sequences --folder_gt testsets/vimeo90k/vimeo_septuplet/sequences --tile 0 0 0 --tile_overlap 0 20 200". However, I cannot download Vimeo-90k and when I run this program, it raises error like this: /home/dell/anaconda3/envs/test/lib/python3.9/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at /opt/conda/conda-bld/pytorch_1678402374358/work/aten/src/ATen/native/TensorShape.cpp:3483.) return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined] loading model from ./model_zoo/rvrt/model_zoo/rvrt/002_RVRT_videosr_bi_Vimeo_14frames.pth Vimeo dataset is not at testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences! Please refer to #training of Readme.md to download it. 4 5 6 7 [ WARN:0@3.707] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0266/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.707] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0268/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.707] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0275/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.707] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0278/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.708] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0285/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.708] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0287/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.708] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0291/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.710] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0619/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.724] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0622/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.724] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0625/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.724] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0627/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.724] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0628/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.724] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0629/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.725] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0632/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.725] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0786/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.725] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0646/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.725] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0791/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.725] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0794/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.725] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0797/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.725] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0783/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.725] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0790/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.725] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0800/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.725] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0643/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.725] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0804/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.726] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0807/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.726] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0808/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.726] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0636/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.726] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0805/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.726] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septupletmatlabLRx4/sequences/00001/0799/im1.png'): can't open/read file: check file path/integrity [ WARN:0@3.726] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septuplet_matlabLRx4/sequences/00001/0812/im1.png'): can't open/read file: check file path/integrity Traceback (most recent call last): File "/home/dell/RVRT-main(2)/main_testrvrt.py", line 362, in [ WARN:0@3.727] global loadsave.cpp:244 findDecoder imread('testsets/vimeo90k/vimeo_septuplet_matlabLRx4/sequences/00001/0809/im1.png'): can't open/read file: check file path/integrity main() File "/home/dell/RVRT-main(2)/main_test_rvrt.py", line 89, in main for idx, batch in enumerate(test_loader): File "/home/dell/anaconda3/envs/test/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 634, in next data = self._next_data() File "/home/dell/anaconda3/envs/test/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 1346, in _next_data return self._process_data(data) File "/home/dell/anaconda3/envs/test/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 1372, in _process_data data.reraise() File "/home/dell/anaconda3/envs/test/lib/python3.9/site-packages/torch/_utils.py", line 644, in reraise raise exception AttributeError: Caught AttributeError in DataLoader worker process 0. Original Traceback (most recent call last): File "/home/dell/anaconda3/envs/test/lib/python3.9/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop data = fetcher.fetch(index) File "/home/dell/anaconda3/envs/test/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/dell/anaconda3/envs/test/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 51, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/dell/RVRT-main(2)/data/dataset_video_test.py", line 278, in getitem imgs_lq = utils_video.read_img_seq(lq_path) File "/home/dell/RVRT-main(2)/utils/utils_video.py", line 75, in read_img_seq imgs = [cv2.imread(v).astype(np.float32) / 255. for v in img_paths] File "/home/dell/RVRT-main(2)/utils/utils_video.py", line 75, in imgs = [cv2.imread(v).astype(np.float32) / 255. for v in img_paths] AttributeError: 'NoneType' object has no attribute 'astype'

How can I solve this issue? I really appreciate someone's reply!

r3a1d3a1 commented 1 year ago

https://github.com/JingyunLiang/RVRT/blob/a5d406cd77c687774c5553bc57048e92b161829b/main_test_rvrt.py#L160