data_paths.py 1.25 KB
Newer Older
yangzhong's avatar
v1.0  
yangzhong committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#IMAGE_FOLDER_DICT = {
IMAGE_FOLDER_DICT_GCP = {
    #"LLaVA-Pretrain": "/public/opendas/DL_DATA/LLaVA-Pretrain",
    # "ai2d": "/export/home/blip3_data/ocr_datasets/ai2d",
    # "dvqa": "/export/home/blip3_data/ocr_datasets/DVQA",
    # "docvqa": "/export/home/blip3_data/ocr_datasets/DocVQA", # Put this before vg, bc docvqa files contain characters. 
    # "ChartQA_Dataset": "/export/home/blip3_data/ocr_datasets/chartQA/ChartQA_Dataset",
    "coco/som_train2017": "/blip-3_pytorch/dataset/SoM-LLaVA/som_train2017",
    # "coco/train2017": "/export/home/blip3_data/coco/images/train2017",
    # "ocr_vqa": "/export/home/blip3_data/ocr_vqa",
    # "vg": "/export/home/blip3_data/visual-genome",
    # "gqa": "/export/home/blip3_data/GQA",
    # "share_textvqa": "/export/home/blip3_data/share_textvqa", # Put this before the substring below.
    # "textvqa": "/export/home/blip3_data/TextVQA",
    # 'wikiart': "/export/home/blip3_data/wikiart",
    # 'sam/images': '/export/home/blip3_data/sam/images',
    # "web-celebrity": "/export/home/blip3_data/web-celebrity",
    # "web-landmark": "/export/home/blip3_data/web-landmark",
    # "llava/llava_pretrain": "/export/home/blip3_data/llava/llava_pretrain",
    # "train2017": "/export/home/blip3_data/coco/images/train2017",
}