From ce61a6781cbf34b699be773cdc25dabe6045fb40 Mon Sep 17 00:00:00 2001 From: chaoyi-wu <18706207256@163.com> Date: Sun, 10 Nov 2024 14:57:28 +0800 Subject: [PATCH] Update multi_dataset_test_for_close.py --- src/Dataset/multi_dataset_test_for_close.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/Dataset/multi_dataset_test_for_close.py b/src/Dataset/multi_dataset_test_for_close.py index f6cb29a..f5e127c 100644 --- a/src/Dataset/multi_dataset_test_for_close.py +++ b/src/Dataset/multi_dataset_test_for_close.py @@ -158,12 +158,12 @@ def __init__(self, text_tokenizer, test_split = 'close', max_seq = 2048, max_img self.data_whole_2D = self.data_whole_2D + [{'medpix_single_dataset':i} for i in range(len(medpix_single_dataset))] print('medpix_single_dataset loaded') - vqarad_dataset = PMCVQA_Dataset(csv_path = '/gpfs/home/cs/leijiayu/wuchaoyi/wangyingjie/src/New_Dataset/data_csv/vqarad_test.csv') + vqarad_dataset = VQA_Dataset(csv_path = '/gpfs/home/cs/leijiayu/wuchaoyi/wangyingjie/src/New_Dataset/data_csv/vqarad_test.csv') self.dataset_reflect['vqarad_dataset'] = vqarad_dataset self.data_whole_2D = self.data_whole_2D + [{'vqarad_dataset':i} for i in range(len(vqarad_dataset))] print('vqarad_dataset loaded') - slake_dataset = PMCVQA_Dataset(csv_path = '/gpfs/home/cs/leijiayu/wuchaoyi/wangyingjie/src/New_Dataset/data_csv/slakevqa_test.csv') + slake_dataset = VQA_Dataset(csv_path = '/gpfs/home/cs/leijiayu/wuchaoyi/wangyingjie/src/New_Dataset/data_csv/slakevqa_test.csv') self.dataset_reflect['slake_dataset'] = slake_dataset self.data_whole_2D = self.data_whole_2D + [{'slake_dataset':i} for i in range(len(slake_dataset))] print('slake_dataset loaded') @@ -302,4 +302,4 @@ def text_add_image(self,images,question,answer): # print(len(dataset)) # for i in range(10): # dataset[i] -# input() \ No newline at end of file +# input()