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

75
mmpretrain/datasets/scienceqa.py
View File

@ -62,42 +62,43 @@ 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, 'question':
'question': ann['question'],
ann['question'], 'choices':
'choices': ann['choices'],
ann['choices'], 'answer':
'answer': ann['answer'],
ann['answer'], 'hint':
'hint': ann['hint'],
ann['hint'], 'image_name':
'image_name': ann['image'],
ann['image'], 'task':
'task': ann['task'],
ann['task'], 'grade':
'grade': ann['grade'],
ann['grade'], 'subject':
'subject': ann['subject'],
ann['subject'], 'topic':
'topic': ann['topic'],
ann['topic'], 'category':
'category': ann['category'],
ann['category'], 'skill':
'skill': ann['skill'],
ann['skill'], 'lecture':
'lecture': ann['lecture'],
ann['lecture'], 'solution':
'solution': ann['solution'],
ann['solution'], 'split':
'split': ann['split'],
ann['split'], 'img_path':
'img_path': file_backend.join_path(img_prefix, data_id, ann['image'])
file_backend.join_path(img_prefix, data_id, if ann['image'] is not None else None,
ann['image']), # noqa 'caption':
} ann['caption'],
data_list.append(data_info) }
data_list.append(data_info)
return data_list return data_list