line_net.yaml 846 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. image:
  2. mean: [109.730, 103.832, 98.681]
  3. stddev: [22.275, 22.124, 23.229]
  4. batch_size: 4
  5. batch_size_eval: 2
  6. # backbone multi-task parameters
  7. head_size: [[2], [1], [2]]
  8. loss_weight:
  9. jmap: 8.0
  10. lmap: 0.5
  11. joff: 0.25
  12. lpos: 1
  13. lneg: 1
  14. boxes: 1.0
  15. # backbone parameters
  16. backbone: resnet50_fpn
  17. # backbone: unet
  18. depth: 4
  19. num_stacks: 1
  20. num_blocks: 1
  21. num_classes: 2
  22. # sampler parameters
  23. ## static sampler
  24. n_stc_posl: 300
  25. n_stc_negl: 40
  26. ## dynamic sampler
  27. n_dyn_junc: 300
  28. n_dyn_posl: 300
  29. n_dyn_negl: 80
  30. n_dyn_othr: 600
  31. # LOIPool layer parameters
  32. n_pts0: 32
  33. n_pts1: 8
  34. # line verification network parameters
  35. dim_loi: 128
  36. dim_fc: 1024
  37. # maximum junction and line outputs
  38. n_out_junc: 250
  39. n_out_line: 2500
  40. # additional ablation study parameters
  41. use_cood: 0
  42. use_slop: 0
  43. use_conv: 0
  44. # junction threashold for evaluation (See #5)
  45. eval_junc_thres: 0.008