-
Notifications
You must be signed in to change notification settings - Fork 72
/
Base_Omni3D.yaml
18 lines (18 loc) · 1.07 KB
/
Base_Omni3D.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
_BASE_: "Base.yaml"
SOLVER:
TYPE: "sgd"
IMS_PER_BATCH: 192
BASE_LR: 0.12
STEPS: (69600, 92800)
MAX_ITER: 116000
WARMUP_ITERS: 3625
TEST:
EVAL_PERIOD: 29000
VIS_PERIOD: 2320
DATASETS:
TRAIN: ('SUNRGBD_train', 'SUNRGBD_val', 'Hypersim_train', 'Hypersim_val', 'ARKitScenes_train', 'ARKitScenes_val', 'Objectron_train', 'Objectron_val', 'nuScenes_train', 'nuScenes_val', 'KITTI_train', 'KITTI_val')
TEST: ('SUNRGBD_test', 'Hypersim_test', 'ARKitScenes_test', 'Objectron_test', 'KITTI_test', 'nuScenes_test')
CATEGORY_NAMES: ('chair', 'table', 'cabinet', 'car', 'lamp', 'books', 'sofa', 'pedestrian', 'picture', 'window', 'pillow', 'truck', 'door', 'blinds', 'sink', 'shelves', 'television', 'shoes', 'cup', 'bottle', 'bookcase', 'laptop', 'desk', 'cereal box', 'floor mat', 'traffic cone', 'mirror', 'barrier', 'counter', 'camera', 'bicycle', 'toilet', 'bus', 'bed', 'refrigerator', 'trailer', 'box', 'oven', 'clothes', 'van', 'towel', 'motorcycle', 'night stand', 'stove', 'machine', 'stationery', 'bathtub', 'cyclist', 'curtain', 'bin')
MODEL:
ROI_HEADS:
NUM_CLASSES: 50