[Fix] Align COCO dataset format.

open-mmlab · Jun 19, 2023 · dbef2b4 · dbef2b4
1 parent d6056af
commit dbef2b4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/configs/_base_/datasets/coco_okvqa.py b/configs/_base_/datasets/coco_okvqa.py
@@ -44,7 +44,7 @@
     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 @@
     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',

diff --git a/configs/minigpt4/minigpt-4_vicuna-7b_caption.py b/configs/minigpt4/minigpt-4_vicuna-7b_caption.py
@@ -14,9 +14,7 @@
     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