forked from autonomousvision/transfuser
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.py
32 lines (24 loc) · 1.01 KB
/
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
import os
class GlobalConfig:
# Data
seq_len = 1 # input timesteps
pred_len = 0 # future waypoints predicted, not required for CILRS
root_dir = '/is/rg/avg/aprakash/carla9-10_data/opengl/all_towns_data'
train_towns = ['Town01', 'Town02', 'Town03', 'Town04', 'Town06', 'Town07', 'Town10']
val_towns = ['Town05']
train_data, val_data = [], []
for town in train_towns:
train_data.append(os.path.join(root_dir, town+'_tiny'))
train_data.append(os.path.join(root_dir, town+'_short'))
for town in val_towns:
val_data.append(os.path.join(root_dir, town+'_short'))
ignore_sides = True # don't consider side cameras
ignore_rear = True # don't consider rear cameras
input_resolution = 256
scale = 1 # image pre-processing
crop = 256 # image pre-processing
max_throttle = 0.75 # upper limit on throttle signal value in dataset
lr = 1e-4 # learning rate
def __init__(self, **kwargs):
for k,v in kwargs.items():
setattr(self, k, v)