1
1
# Copyright 2021-2022 Toyota Research Institute. All rights reserved.
2
2
import os
3
- from collections import OrderedDict
4
3
5
- __version__ = '1.0'
4
+ __version__ = "1.6.0"
6
5
7
- DGP_PATH = os .getenv (' DGP_PATH' , default = os .getenv (' HOME' , os .getcwd ()))
8
- DGP_DATA_DIR = os .path .join (DGP_PATH , ' .dgp' )
9
- DGP_CACHE_DIR = os .path .join (DGP_DATA_DIR , ' cache' )
10
- DGP_DATASETS_CACHE_DIR = os .path .join (DGP_DATA_DIR , ' datasets' )
6
+ DGP_PATH = os .getenv (" DGP_PATH" , default = os .getenv (" HOME" , os .getcwd ()))
7
+ DGP_DATA_DIR = os .path .join (DGP_PATH , " .dgp" )
8
+ DGP_CACHE_DIR = os .path .join (DGP_DATA_DIR , " cache" )
9
+ DGP_DATASETS_CACHE_DIR = os .path .join (DGP_DATA_DIR , " datasets" )
11
10
12
11
TRI_DGP_FOLDER_PREFIX = "dgp/"
13
12
TRI_RAW_FOLDER_PREFIX = "raw/"
14
13
TRI_DGP_JSON_PREFIX = "dataset_v"
15
14
16
15
# DGP Directory structure constants
17
- RGB_FOLDER = ' rgb'
18
- POINT_CLOUD_FOLDER = ' point_cloud'
16
+ RGB_FOLDER = " rgb"
17
+ POINT_CLOUD_FOLDER = " point_cloud"
19
18
RADAR_POINT_CLOUD_FOLDER = "radar_point_cloud"
20
- BOUNDING_BOX_2D_FOLDER = ' bounding_box_2d'
21
- BOUNDING_BOX_3D_FOLDER = ' bounding_box_3d'
22
- SEMANTIC_SEGMENTATION_2D_FOLDER = ' semantic_segmentation_2d'
23
- SEMANTIC_SEGMENTATION_3D_FOLDER = ' semantic_segmentation_3d'
24
- INSTANCE_SEGMENTATION_2D_FOLDER = ' instance_segmentation_2d'
25
- INSTANCE_SEGMENTATION_3D_FOLDER = ' instance_segmentation_3d'
26
- DEPTH_FOLDER = ' depth'
19
+ BOUNDING_BOX_2D_FOLDER = " bounding_box_2d"
20
+ BOUNDING_BOX_3D_FOLDER = " bounding_box_3d"
21
+ SEMANTIC_SEGMENTATION_2D_FOLDER = " semantic_segmentation_2d"
22
+ SEMANTIC_SEGMENTATION_3D_FOLDER = " semantic_segmentation_3d"
23
+ INSTANCE_SEGMENTATION_2D_FOLDER = " instance_segmentation_2d"
24
+ INSTANCE_SEGMENTATION_3D_FOLDER = " instance_segmentation_3d"
25
+ DEPTH_FOLDER = " depth"
27
26
EXTRA_DATA_FOLDER = "extra_data"
28
27
FEATURE_ONTOLOGY_FOLDER = "feature_ontology"
29
28
AGENT_FOLDER = "agent"
30
29
CLASSIFICATION_FOLDER = "classification"
31
30
32
31
# Scene Directory structure constants
33
- CALIBRATION_FOLDER = ' calibration'
34
- ONTOLOGY_FOLDER = ' ontology'
35
- SCENE_JSON_FILENAME = ' scene.json'
32
+ CALIBRATION_FOLDER = " calibration"
33
+ ONTOLOGY_FOLDER = " ontology"
34
+ SCENE_JSON_FILENAME = " scene.json"
36
35
37
36
# DGP file naming conventions
38
37
TRI_DGP_SCENE_DATASET_JSON_NAME = "scene_dataset_v{version}.json"
39
38
TRI_DGP_SCENE_JSON_NAME = "scene_{scene_hash}.json"
40
- ANNOTATION_FILE_NAME = ' {image_content_hash}_{annotation_content_hash}.json'
39
+ ANNOTATION_FILE_NAME = " {image_content_hash}_{annotation_content_hash}.json"
41
40
42
41
# DGP file naming conventions
43
42
TRI_DGP_SCENE_DATASET_JSON_NAME = "scene_dataset_v{version}.json"
44
43
TRI_DGP_AGENT_TRACKS_JSON_NAME = "agent_tracks_{track_hash}.json"
45
44
TRI_DGP_SCENE_JSON_NAME = "scene_{scene_hash}.json"
46
- ANNOTATION_FILE_NAME = ' {image_content_hash}_{annotation_content_hash}.json'
45
+ ANNOTATION_FILE_NAME = " {image_content_hash}_{annotation_content_hash}.json"
47
46
TRI_DGP_AGENTS_JSON_NAME = "agents_{agent_hash}.json"
48
47
TRI_DGP_AGENTS_SLICES_JSON_NAME = "agents_slices_{slice_hash}.json"
49
48
50
49
# Autolabel constants
51
- AUTOLABEL_FOLDER = ' autolabels'
52
- AUTOLABEL_SCENE_JSON_NAME = ' scene.json'
50
+ AUTOLABEL_FOLDER = " autolabels"
51
+ AUTOLABEL_SCENE_JSON_NAME = " scene.json"
0 commit comments