[Fix] Align COCO dataset format.

pull/1653/head
mzr1996 2023-06-19 07:24:07 +00:00
parent d6056af2b8
commit dbef2b41c6
2 changed files with 3 additions and 5 deletions

View File

@ -44,7 +44,7 @@ train_dataloader = dict(
dataset=dict(
type='COCOVQA',
data_root='data/coco',
data_prefix='images/train2014',
data_prefix='train2014',
question_file=
'annotations/okvqa_OpenEnded_mscoco_train2014_questions.json',
ann_file='annotations/okvqa_mscoco_train2014_annotations.json',
@ -60,7 +60,7 @@ val_dataloader = dict(
dataset=dict(
type='COCOVQA',
data_root='data/coco',
data_prefix='images/val2014',
data_prefix='val2014',
question_file=
'annotations/okvqa_OpenEnded_mscoco_val2014_questions.json',
ann_file='annotations/okvqa_mscoco_val2014_annotations.json',

View File

@ -14,9 +14,7 @@ test_pipeline = [
dict(type='PackInputs', meta_keys=['image_id']),
]
val_dataloader = dict(
batch_size=1,
dataset=dict(data_prefix=dict(img_path='images'), pipeline=test_pipeline))
val_dataloader = dict(batch_size=1, dataset=dict(pipeline=test_pipeline))
test_dataloader = val_dataloader
# model settings