Chengfang Lu
7c43b44c57
Huaxu Sentinel Active Safety Platform with embedded algorithm code, Docker Compose setup, and vendored dataset scaffolds for clone-and-run. Co-authored-by: Cursor <cursoragent@cursor.com>
46 lines
587 B
Python
Executable File
46 lines
587 B
Python
Executable File
# DATA
|
|
dataset = 'CULane'
|
|
data_root = 'C:\\data\\Tusimple\\test_set'
|
|
|
|
# TRAIN
|
|
epoch = 50
|
|
batch_size = 32
|
|
optimizer = 'SGD' #['SGD','Adam']
|
|
learning_rate = 0.1
|
|
weight_decay = 1e-4
|
|
momentum = 0.9
|
|
|
|
scheduler = 'multi' #['multi', 'cos']
|
|
steps = [25,38]
|
|
gamma = 0.1
|
|
warmup = 'linear'
|
|
warmup_iters = 695
|
|
|
|
# NETWORK
|
|
use_aux = True
|
|
griding_num = 200
|
|
backbone = '18'
|
|
|
|
# LOSS
|
|
sim_loss_w = 0.0
|
|
shp_loss_w = 0.0
|
|
|
|
# EXP
|
|
note = ''
|
|
|
|
log_path = None
|
|
|
|
# FINETUNE or RESUME MODEL PATH
|
|
finetune = None
|
|
resume = None
|
|
|
|
# TEST
|
|
test_model = './model/culane_18.pth'
|
|
test_work_dir = './tmp'
|
|
|
|
num_lanes = 4
|
|
|
|
|
|
|
|
|