42 lines
644 B
Python
42 lines
644 B
Python
|
# 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
|