From dbef2b41c6845e86b68370287537924d46fd41d2 Mon Sep 17 00:00:00 2001
From: mzr1996 <mzr1996@163.com>
Date: Mon, 19 Jun 2023 07:24:07 +0000
Subject: [PATCH] [Fix] Align COCO dataset format.

---
 configs/_base_/datasets/coco_okvqa.py           | 4 ++--
 configs/minigpt4/minigpt-4_vicuna-7b_caption.py | 4 +---
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/configs/_base_/datasets/coco_okvqa.py b/configs/_base_/datasets/coco_okvqa.py
index 0c30fda6..16f1577d 100644
--- a/configs/_base_/datasets/coco_okvqa.py
+++ b/configs/_base_/datasets/coco_okvqa.py
@@ -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',
diff --git a/configs/minigpt4/minigpt-4_vicuna-7b_caption.py b/configs/minigpt4/minigpt-4_vicuna-7b_caption.py
index c60b63e9..704760af 100644
--- a/configs/minigpt4/minigpt-4_vicuna-7b_caption.py
+++ b/configs/minigpt4/minigpt-4_vicuna-7b_caption.py
@@ -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