From 02d1c3f2e9d9c980e0a4fe64d6e1253e1244e157 Mon Sep 17 00:00:00 2001 From: hmtbgc Date: Tue, 12 Sep 2023 21:47:16 +0800 Subject: [PATCH] restore files deleted by mistake --- mmpretrain/datasets/__init__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mmpretrain/datasets/__init__.py b/mmpretrain/datasets/__init__.py index 59d970da..e621e157 100644 --- a/mmpretrain/datasets/__init__.py +++ b/mmpretrain/datasets/__init__.py @@ -41,6 +41,8 @@ if WITH_MULTIMODAL: from .flickr30k_caption import Flickr30kCaption from .flickr30k_retrieval import Flickr30kRetrieval from .gqa_dataset import GQA + from .iconqa import IconQA + from .infographic_vqa import InfographicVQA from .minigpt4_dataset import MiniGPT4Dataset from .nocaps import NoCaps from .ocr_vqa import OCRVQA @@ -55,5 +57,6 @@ if WITH_MULTIMODAL: 'COCOCaption', 'COCORetrieval', 'COCOVQA', 'FlamingoEvalCOCOCaption', 'FlamingoEvalCOCOVQA', 'Flickr30kCaption', 'Flickr30kRetrieval', 'RefCOCO', 'VisualGenomeQA', 'ScienceQA', 'NoCaps', 'GQA', 'TextVQA', - 'VSR', 'VizWiz', 'OCRVQA', 'MiniGPT4Dataset' + 'VSR', 'VizWiz', 'OCRVQA', 'InfographicVQA', 'IconQA', + 'MiniGPT4Dataset' ])