wireframe.yaml 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. io:
  2. logdir: logs/
  3. datadir: I:/datasets/wirenet_1000
  4. resume_from:
  5. num_workers: 8
  6. tensorboard_port: 6000
  7. validation_interval: 300
  8. model:
  9. image:
  10. mean: [109.730, 103.832, 98.681]
  11. stddev: [22.275, 22.124, 23.229]
  12. batch_size: 4
  13. batch_size_eval: 2
  14. # backbone multi-task parameters
  15. head_size: [[2], [1], [2]]
  16. loss_weight:
  17. jmap: 8.0
  18. lmap: 0.5
  19. joff: 0.25
  20. lpos: 1
  21. lneg: 1
  22. boxes: 1.0
  23. # backbone parameters
  24. backbone: fasterrcnn_resnet50
  25. # backbone: unet
  26. depth: 4
  27. num_stacks: 1
  28. num_blocks: 1
  29. # sampler parameters
  30. ## static sampler
  31. n_stc_posl: 300
  32. n_stc_negl: 40
  33. ## dynamic sampler
  34. n_dyn_junc: 300
  35. n_dyn_posl: 300
  36. n_dyn_negl: 80
  37. n_dyn_othr: 600
  38. # LOIPool layer parameters
  39. n_pts0: 32
  40. n_pts1: 8
  41. # line verification network parameters
  42. dim_loi: 128
  43. dim_fc: 1024
  44. # maximum junction and line outputs
  45. n_out_junc: 250
  46. n_out_line: 2500
  47. # additional ablation study parameters
  48. use_cood: 0
  49. use_slop: 0
  50. use_conv: 0
  51. # junction threashold for evaluation (See #5)
  52. eval_junc_thres: 0.008
  53. optim:
  54. name: Adam
  55. lr: 4.0e-4
  56. amsgrad: True
  57. weight_decay: 1.0e-4
  58. max_epoch: 1000
  59. lr_decay_epoch: 10