-
Notifications
You must be signed in to change notification settings - Fork 41
/
dataset_config.py
52 lines (43 loc) · 1.16 KB
/
dataset_config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
Pretrain = {
"chat_path": "${PATH}/CC3M-595K/chat.json",
"CC3M": "${PATH}/CC3M-595K",
}
VIT = {
"chat_path": "${PATH}/llava_instruct_150k.json",
"COCO2017": "${PATH}/COCO2017/train2017",
}
MIMIC_imageonly = {
"chat_path": "${PATH}/MIMIC-IT-imageonly.json",
"CDG": "${PATH}/CGD/images",
"LA": "${PATH}/LA/images",
"SD": "${PATH}/SD/images",
}
COCO_CAP = {
"chat_path": "${PATH}/COCO/coco_cap_chat.json",
"COCO2014": "${PATH}/COCO2014/train2014",
}
COCO_REG = {
"chat_path": "${PATH}/COCO/coco_reg_chat.json",
"COCO2014": "${PATH}/COCO2014/train2014",
}
COCO_REC = {
"chat_path": "${PATH}/COCO/coco_rec_chat.json",
"COCO2014": "${PATH}/COCO2014/train2014",
}
VIDEO = {
"chat_path": "${PATH}/video_chat.json",
"VIDEO": "${PATH}/Activity_Videos",
}
SQA = {
"chat_path": "${PATH}/llava_train_QCM-LEA.json",
"ScienceQA": "${PATH}/scienceqa/train",
}
Pretrain_valley_llava = {
"chat_path": "${PATH}/valley_llavaimage.json",
"valley": "${PATH}/Data",
"llava": "${PATH}/Data", # from llava v1.5
}
LLaVA = {
"chat_path": "${PATH}/llavaimage_tune.json",
"llava": "${PATH}/Data", # from llava v1.5
}