[Fix] fix test of Vid4 datasets bug (#1293)

* [Fix] fix test of Vid4 datasets bug * Update tof_x4_official_vimeo90k.py * Update tdan_x4_1xb16-lr1e-4-400k_vimeo90k-bd.py * Update tdan_test_config.py
open-mmlab · Oct 14, 2022 · 2547173 · 2547173
1 parent e25023b
commit 2547173
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 8 deletions.
diff --git a/configs/_base_/datasets/basicvsr_test_config.py b/configs/_base_/datasets/basicvsr_test_config.py
@@ -109,7 +109,6 @@
         metainfo=dict(dataset_type='udm10', task_name='vsr'),
         data_root=udm10_data_root,
         data_prefix=dict(img='BDx4', gt='GT'),
-        num_input_frames=15,
         pipeline=udm10_pipeline))
 
 udm10_evaluator = [
@@ -138,8 +137,6 @@
         data_prefix=dict(img='BDx4', gt='GT'),
         ann_file='meta_info_Vid4_GT.txt',
         depth=1,
-        num_input_frames=7,
-        fixed_seq_len=7,
         pipeline=vid4_pipeline))
 
 vid4_bi_dataloader = dict(
@@ -154,8 +151,6 @@
         data_prefix=dict(img='BIx4', gt='GT'),
         ann_file='meta_info_Vid4_GT.txt',
         depth=1,
-        num_input_frames=7,
-        fixed_seq_len=7,
         pipeline=vid4_pipeline))
 
 vid4_bd_evaluator = [

diff --git a/configs/basicvsr/basicvsr_2xb4_vimeo90k-bd.py b/configs/basicvsr/basicvsr_2xb4_vimeo90k-bd.py
@@ -68,8 +68,6 @@
         data_prefix=dict(img='BDx4', gt='GT'),
         ann_file='meta_info_Vid4_GT.txt',
         depth=1,
-        num_input_frames=7,
-        fixed_seq_len=7,
         pipeline=val_pipeline))
 
 val_evaluator = [

diff --git a/configs/basicvsr_pp/basicvsr-pp_c128n25_600k_ntire-decompress-track1.py b/configs/basicvsr_pp/basicvsr-pp_c128n25_600k_ntire-decompress-track1.py
@@ -52,7 +52,6 @@
         metainfo=dict(dataset_type='ntire21_track1', task_name='vsr'),
         data_root='data/NTIRE21_decompression_track1',
         data_prefix=dict(img='LQ', gt='GT'),
-        num_input_frames=15,
         pipeline=test_pipeline))
 
 test_evaluator = [

diff --git a/mmedit/datasets/transforms/generate_frame_indices.py b/mmedit/datasets/transforms/generate_frame_indices.py
@@ -260,6 +260,8 @@ def transform(self, results):
         self.sequence_length = results['sequence_length']
         num_input_frames = results.get('num_input_frames',
                                        self.sequence_length)
+        if num_input_frames is None:
+            num_input_frames = self.sequence_length
 
         # randomly select a frame as start
         if self.sequence_length - num_input_frames * interval < 0: