Spaces:
Running
Running
| ### [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 |