feat: initial HSAP platform
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>
This commit is contained in:
41
algorithms/lane_ufld/code/UFLD/configs/tusimple_res18.py
Executable file
41
algorithms/lane_ufld/code/UFLD/configs/tusimple_res18.py
Executable file
@@ -0,0 +1,41 @@
|
||||
# DATA
|
||||
dataset = 'Tusimple'
|
||||
data_root = '/mnt/HDisk2T/liuxy51/ganxian/data_luojk'
|
||||
|
||||
# TRAIN
|
||||
epoch = 500 # 10
|
||||
batch_size = 32 # 4
|
||||
optimizer = 'Adam' #['SGD','Adam']
|
||||
learning_rate = 1e-5
|
||||
weight_decay = 1e-4
|
||||
momentum = 0.9
|
||||
scheduler = 'cos' #['multi', 'cos']
|
||||
|
||||
# steps = [50,75]
|
||||
gamma = 0.1
|
||||
warmup = 'linear'
|
||||
warmup_iters = 100
|
||||
|
||||
# NETWORK
|
||||
backbone = '18'
|
||||
griding_num = 100
|
||||
use_aux = False
|
||||
|
||||
# LOSS
|
||||
sim_loss_w = 1.0
|
||||
shp_loss_w = 0.0
|
||||
|
||||
# EXP
|
||||
note = '_ufld_2lanes_res18'
|
||||
|
||||
log_path = './log'
|
||||
|
||||
# FINETUNE or RESUME MODEL PATH
|
||||
finetune = None
|
||||
resume = None
|
||||
|
||||
# TESTNone
|
||||
test_model = './model/lane_m599_all.pth'
|
||||
test_work_dir = './tmp'
|
||||
|
||||
num_lanes = 2
|
||||
Reference in New Issue
Block a user