data-preprocessing/paths.py at main · DriveFusion/data-preprocessing · GitHub

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
import os

# BDD-X Paths
BDD_X_DATASET_PATH = './datasets/bdd_x_dataset'
BDD_X_TRAINING_VIDEOS_PATH = os.path.join(BDD_X_DATASET_PATH, 'train/videos')

# LingoQA Paths
LINGO_DATASET_PATH = './datasets/lingoqa_dataset'
LINGO_ACTION_PATH = os.path.join(LINGO_DATASET_PATH, 'action')
LINGO_SCENERY_PATH = os.path.join(LINGO_DATASET_PATH, 'scenery')
LINGO_EVAL_PATH = os.path.join(LINGO_DATASET_PATH, 'evaluation')
LINGO_IMAGES_PATH = os.path.join(LINGO_DATASET_PATH, 'images')
LINGO_TRAIN_PATH = os.path.join(LINGO_DATASET_PATH, 'train')
LINGO_VAL_PATH = os.path.join(LINGO_DATASET_PATH, 'val')

# DriveGPT4 Paths
DRIVE_BDDX_DATASET_PATH = './datasets/drivegpt4_dataset'
DRIVE_BDDX_IMAGES_PATH = os.path.join(DRIVE_BDDX_DATASET_PATH, 'BDD_X_imgs_select')
DRIVE_BDDX_VIDEOS_PATH = os.path.join(DRIVE_BDDX_DATASET_PATH, 'videos')

# DriveLM Paths
DRIVELM_DATASET_PATH = './datasets/drivelm_dataset'
DRIVELM_IMAGES_TRAIN_PATH = '/home/cis-g1/Documents/datasets/DriveLM/drivelm_nus_imgs_train/nuscenes/samples'
DRIVELM_IMAGES_WITH_BBOX_PATH = '/home/cis-g1/Documents/datasets/DriveLM/drivelm_nus_imgs_bbox_train/nuscenes/samples'
FINAL_DRIVELM_IMAGES_PATH = '/home/cis-g1/Documents/datasets/DriveLM/drivelm_nus_imgs_final_train/nuscenes/samples'

# Nuinstruct Paths
NUINSTRUCT_DATASET_PATH = './datasets/nuinstruct_dataset'

# Our Paths
WINDOWS_SYSTEM_PATH = 'E:\datasets'
UNI_LINUX_SYSTEM_PATH = '/home/cis-g1/Documents' # UNI_VM
DATACRUNCH_LINUX_SYSTEM_PATH = '/mnt/mydrive'
OUR_DATASET_PATH = './our_datasets'
OUR_LINGO_DATASET_PATH = os.path.join(OUR_DATASET_PATH, 'lingoqa_dataset')
OUR_LINGO_ACTION_DATASET_PATH = os.path.join(OUR_LINGO_DATASET_PATH, 'action')
OUR_LINGO_SCENERY_DATASET_PATH = os.path.join(OUR_LINGO_DATASET_PATH, 'scenery')
OUR_LINGO_EVAL_DATASET_PATH = os.path.join(OUR_LINGO_DATASET_PATH, 'evaluation')
OUR_DRIVE_BDDX_DATASET_PATH = os.path.join(OUR_DATASET_PATH, 'drivegpt4_dataset')
OUR_DRIVE_BDDX_TRAIN_DATASET_PATH = os.path.join(OUR_DRIVE_BDDX_DATASET_PATH, 'train')
OUR_DRIVE_BDDX_TEST_DATASET_PATH = os.path.join(OUR_DRIVE_BDDX_DATASET_PATH, 'test')
OUR_DRIVELM_DATASET_PATH = os.path.join(OUR_DATASET_PATH, 'drivelm_dataset')
OUR_NUINSTRUCT_DATASET_PATH = os.path.join(OUR_DATASET_PATH, 'nuinstruct_dataset')
OUR_NUINSTRUCT_DATASET_TRAIN_PATH = os.path.join(OUR_NUINSTRUCT_DATASET_PATH, 'train')
OUR_NUINSTRUCT_DATASET_VAL_PATH = os.path.join(OUR_NUINSTRUCT_DATASET_PATH, 'val')
OUR_NUINSTRUCT_DATASET_TEST_PATH = os.path.join(OUR_NUINSTRUCT_DATASET_PATH, 'test')