mirrors/mmclassification
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmclassification.git synced 2025-06-03 21:53:55 +08:00
Code Issues Projects Releases Wiki Activity

[Feature]: Add caption

Browse Source
This commit is contained in:
liuyuan 2023-05-19 16:11:01 +08:00
parent 6847d20d57
commit 1b8e86dca6
1 changed files with 38 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mmpretrain/datasets/scienceqa.py
View File

@ -62,7 +62,6 @@ class ScienceQA(BaseDataset):
data_list = [] data_list = []
for data_id in current_data_split: for data_id in current_data_split:
ann = annotations[data_id] ann = annotations[data_id]
if ann['image'] is not None:
data_info = { data_info = {
'image_id': 'image_id':
data_id, data_id,
@ -95,8 +94,10 @@ class ScienceQA(BaseDataset):
'split': 'split':
ann['split'], ann['split'],
'img_path': 'img_path':
file_backend.join_path(img_prefix, data_id, file_backend.join_path(img_prefix, data_id, ann['image'])
ann['image']), # noqa if ann['image'] is not None else None,
'caption':
ann['caption'],
} }
data_list.append(data_info) data_list.append(data_info)