ISubGVQA.datasets.build

Functions

build_datasets([args, ans2label_path, label2ans_path])

Build and return datasets for training, validation, and test development.

Module Contents

ISubGVQA.datasets.build.build_datasets(args=None, ans2label_path='./ISubGVQA/meta_info/trainval_ans2label.json', label2ans_path='./ISubGVQA/meta_info/trainval_label2ans.json')

Build and return datasets for training, validation, and test development. Args:

args (Namespace, optional): Arguments containing configuration for dataset building. ans2label_path (str, optional): Path to the JSON file mapping answers to labels. Defaults to “./ISubGVQA/meta_info/trainval_ans2label.json”. label2ans_path (str, optional): Path to the JSON file mapping labels to answers. Defaults to “./ISubGVQA/meta_info/trainval_label2ans.json”.

Returns:

dict: A dictionary containing DataLoader objects for ‘train’, ‘dev’, and ‘testdev’ datasets.