infer_cfg.yml 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108
  1. mode: fluid
  2. draw_threshold: 0.5
  3. metric: COCO
  4. use_dynamic_shape: false
  5. arch: PicoDet
  6. min_subgraph_size: 3
  7. Preprocess:
  8. - interp: 2
  9. keep_ratio: false
  10. target_size:
  11. - 192
  12. - 192
  13. type: Resize
  14. - type: Permute
  15. - stride: 32
  16. type: PadStride
  17. label_list:
  18. - person
  19. - bicycle
  20. - car
  21. - motorcycle
  22. - airplane
  23. - bus
  24. - train
  25. - truck
  26. - boat
  27. - traffic light
  28. - fire hydrant
  29. - stop sign
  30. - parking meter
  31. - bench
  32. - bird
  33. - cat
  34. - dog
  35. - horse
  36. - sheep
  37. - cow
  38. - elephant
  39. - bear
  40. - zebra
  41. - giraffe
  42. - backpack
  43. - umbrella
  44. - handbag
  45. - tie
  46. - suitcase
  47. - frisbee
  48. - skis
  49. - snowboard
  50. - sports ball
  51. - kite
  52. - baseball bat
  53. - baseball glove
  54. - skateboard
  55. - surfboard
  56. - tennis racket
  57. - bottle
  58. - wine glass
  59. - cup
  60. - fork
  61. - knife
  62. - spoon
  63. - bowl
  64. - banana
  65. - apple
  66. - sandwich
  67. - orange
  68. - broccoli
  69. - carrot
  70. - hot dog
  71. - pizza
  72. - donut
  73. - cake
  74. - chair
  75. - couch
  76. - potted plant
  77. - bed
  78. - dining table
  79. - toilet
  80. - tv
  81. - laptop
  82. - mouse
  83. - remote
  84. - keyboard
  85. - cell phone
  86. - microwave
  87. - oven
  88. - toaster
  89. - sink
  90. - refrigerator
  91. - book
  92. - clock
  93. - vase
  94. - scissors
  95. - teddy bear
  96. - hair drier
  97. - toothbrush
  98. NMS:
  99. keep_top_k: 100
  100. name: MultiClassNMS
  101. nms_threshold: 0.5
  102. nms_top_k: 1000
  103. score_threshold: 0.3
  104. fpn_stride:
  105. - 8
  106. - 16
  107. - 32
  108. - 64