wirenet.yaml 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. io:
  2. logdir: logs/
  3. datadir: I:/wirenet_dateset
  4. resume_from:
  5. num_workers: 4
  6. tensorboard_port: 0
  7. validation_interval: 24000
  8. model:
  9. image:
  10. mean: [109.730, 103.832, 98.681]
  11. stddev: [22.275, 22.124, 23.229]
  12. batch_size: 2
  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. # backbone parameters
  23. backbone: stacked_hourglass
  24. depth: 4
  25. num_stacks: 2
  26. num_blocks: 1
  27. # sampler parameters
  28. ## static sampler
  29. n_stc_posl: 300
  30. n_stc_negl: 40
  31. ## dynamic sampler
  32. n_dyn_junc: 300
  33. n_dyn_posl: 300
  34. n_dyn_negl: 80
  35. n_dyn_othr: 600
  36. # LOIPool layer parameters
  37. n_pts0: 32
  38. n_pts1: 8
  39. # line verification network parameters
  40. dim_loi: 128
  41. dim_fc: 1024
  42. # maximum junction and line outputs
  43. n_out_junc: 250
  44. n_out_line: 2500
  45. # additional ablation study parameters
  46. use_cood: 0
  47. use_slop: 0
  48. use_conv: 0
  49. # junction threashold for evaluation (See #5)
  50. eval_junc_thres: 0.008
  51. optim:
  52. name: Adam
  53. lr: 4.0e-4
  54. amsgrad: True
  55. weight_decay: 1.0e-4
  56. max_epoch: 24
  57. lr_decay_epoch: 10