Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[Fix] fix test of Vid4 datasets bug #1293

Merged
merged 4 commits into from
Oct 14, 2022
Merged

[Fix] fix test of Vid4 datasets bug #1293

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
4d44840
[Fix] fix test of Vid4 datasets bug
Z-Fran Oct 13, 2022
6856589
Update tof_x4_official_vimeo90k.py
Z-Fran Oct 13, 2022
3fa5047
Update tdan_x4_1xb16-lr1e-4-400k_vimeo90k-bd.py
Z-Fran Oct 13, 2022
867e5e8
Update tdan_test_config.py
Z-Fran Oct 13, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 0 additions & 5 deletions configs/_base_/datasets/basicvsr_test_config.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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 = [
Expand Down Expand Up @@ -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(
Expand All @@ -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 = [
Expand Down
2 changes: 0 additions & 2 deletions configs/_base_/datasets/tdan_test_config.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,6 @@
data_prefix=dict(img='BDx4', gt='GT'),
ann_file='meta_info_Vid4_GT.txt',
depth=2,
num_input_frames=5,
pipeline=vid4_pipeline))

vid4_bi_dataloader = dict(
Expand All @@ -84,7 +83,6 @@
data_prefix=dict(img='BIx4', gt='GT'),
ann_file='meta_info_Vid4_GT.txt',
depth=2,
num_input_frames=5,
pipeline=vid4_pipeline))

vid4_bd_evaluator = [
Expand Down
2 changes: 0 additions & 2 deletions configs/basicvsr/basicvsr_2xb4_vimeo90k-bd.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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 = [
Expand Down
1 change: 0 additions & 1 deletion configs/basicvsr_pp/basicvsr-pp_c128n25_600k_ntire-decompress-track1.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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 = [
Expand Down
1 change: 0 additions & 1 deletion configs/tdan/tdan_x4_1xb16-lr1e-4-400k_vimeo90k-bd.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,6 @@
data_prefix=dict(img='BDx4', gt='GT'),
ann_file='meta_info_Vid4_GT.txt',
depth=2,
num_input_frames=5,
pipeline=val_pipeline))

# optimizer
Expand Down
1 change: 0 additions & 1 deletion configs/tof/tof_x4_official_vimeo90k.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,6 @@
data_prefix=dict(img='BIx4up_direct', gt='GT'),
ann_file='meta_info_Vid4_GT.txt',
depth=2,
num_input_frames=7,
Z-Fran marked this conversation as resolved.
Show resolved Hide resolved
pipeline=val_pipeline))

# TODO: data is not uploaded yet
Expand Down
2 changes: 2 additions & 0 deletions mmedit/datasets/transforms/generate_frame_indices.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down