File size: 1,071 Bytes
404d2af |
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 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
### [Model parameters]
model_name: "lcnn_simple"
model_architecture: "simple"
# Backbone related config
backbone: "lcnn"
backbone_cfg:
input_channel: 1 # Use RGB images or grayscale images.
depth: 4
num_stacks: 2
num_blocks: 1
num_classes: 5
# Junction decoder related config
junction_decoder: "superpoint_decoder"
junc_decoder_cfg:
# Heatmap decoder related config
heatmap_decoder: "pixel_shuffle"
heatmap_decoder_cfg:
# Shared configurations
grid_size: 8
keep_border_valid: True
# Threshold of junction detection
detection_thresh: 0.0153846 # 1/65
# Threshold of heatmap detection
prob_thresh: 0.5
### [Loss parameters]
weighting_policy: "dynamic"
# [Heatmap loss]
w_heatmap: 0.
w_heatmap_class: 1
heatmap_loss_func: "cross_entropy"
heatmap_loss_cfg:
policy: "dynamic"
# [Junction loss]
w_junc: 0.
junction_loss_func: "superpoint"
junction_loss_cfg:
policy: "dynamic"
### [Training parameters]
learning_rate: 0.0005
epochs: 200
train:
batch_size: 6
num_workers: 8
test:
batch_size: 6
num_workers: 8
disp_freq: 100
summary_freq: 200
max_ckpt: 150 |