11 lines
188 B
Python
11 lines
188 B
Python
BATCH_SIZE = 16
|
|
PROPOSAL_NUM = 6
|
|
CAT_NUM = 4
|
|
INPUT_SIZE = (448, 448) # (w, h)
|
|
LR = 0.001
|
|
WD = 1e-4
|
|
SAVE_FREQ = 1
|
|
resume = ''
|
|
test_model = 'model.ckpt'
|
|
save_dir = '/data_4t/yangz/models/'
|