vqa_data_provider_layer.py 文件源码

python
阅读 22 收藏 0 点赞 0 评论 0

项目:vqa-text 作者: divelab 项目源码 文件源码
def load_vqa_json(data_split):
        """
        Parses the question and answer json files for the given data split.
        Returns the question dictionary and the answer dictionary.
        """
        qdic, adic = {}, {}

        with open(config.DATA_PATHS[data_split]['ques_file'], 'r') as f:
            qdata = json.load(f)['questions']
            for q in qdata:
                qdic[data_split + QID_KEY_SEPARATOR + str(q['question_id'])] = \
                    {'qstr': q['question'], 'iid': q['image_id']}

        if 'test' not in data_split:
            with open(config.DATA_PATHS[data_split]['ans_file'], 'r') as f:
                adata = json.load(f)['annotations']
                for a in adata:
                    adic[data_split + QID_KEY_SEPARATOR + str(a['question_id'])] = \
                        a['answers']

        write_log('parsed ' + str(len(qdic)) + ' questions for ' + data_split, 'log.txt')
        return qdic, adic
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号