config.json 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208
  1. {
  2. "mode": "paddle",
  3. "draw_threshold": 0.5,
  4. "metric": "COCO",
  5. "use_dynamic_shape": false,
  6. "Global": {
  7. "model_name": "RT-DETR-L_wired_table_cell_det"
  8. },
  9. "arch": "DETR",
  10. "min_subgraph_size": 3,
  11. "Preprocess": [
  12. {
  13. "interp": 2,
  14. "keep_ratio": false,
  15. "target_size": [
  16. 640,
  17. 640
  18. ],
  19. "type": "Resize"
  20. },
  21. {
  22. "mean": [
  23. 0.0,
  24. 0.0,
  25. 0.0
  26. ],
  27. "norm_type": "none",
  28. "std": [
  29. 1.0,
  30. 1.0,
  31. 1.0
  32. ],
  33. "type": "NormalizeImage"
  34. },
  35. {
  36. "type": "Permute"
  37. }
  38. ],
  39. "label_list": [
  40. "cell"
  41. ],
  42. "Hpi": {
  43. "backend_configs": {
  44. "paddle_infer": {
  45. "trt_dynamic_shapes": {
  46. "im_shape": [
  47. [
  48. 1,
  49. 2
  50. ],
  51. [
  52. 1,
  53. 2
  54. ],
  55. [
  56. 8,
  57. 2
  58. ]
  59. ],
  60. "image": [
  61. [
  62. 1,
  63. 3,
  64. 640,
  65. 640
  66. ],
  67. [
  68. 1,
  69. 3,
  70. 640,
  71. 640
  72. ],
  73. [
  74. 8,
  75. 3,
  76. 640,
  77. 640
  78. ]
  79. ],
  80. "scale_factor": [
  81. [
  82. 1,
  83. 2
  84. ],
  85. [
  86. 1,
  87. 2
  88. ],
  89. [
  90. 8,
  91. 2
  92. ]
  93. ]
  94. },
  95. "trt_dynamic_shape_input_data": {
  96. "im_shape": [
  97. [
  98. 640,
  99. 640
  100. ],
  101. [
  102. 640,
  103. 640
  104. ],
  105. [
  106. 640,
  107. 640,
  108. 640,
  109. 640,
  110. 640,
  111. 640,
  112. 640,
  113. 640,
  114. 640,
  115. 640,
  116. 640,
  117. 640,
  118. 640,
  119. 640,
  120. 640,
  121. 640
  122. ]
  123. ],
  124. "scale_factor": [
  125. [
  126. 2,
  127. 2
  128. ],
  129. [
  130. 1,
  131. 1
  132. ],
  133. [
  134. 0.67,
  135. 0.67,
  136. 0.67,
  137. 0.67,
  138. 0.67,
  139. 0.67,
  140. 0.67,
  141. 0.67,
  142. 0.67,
  143. 0.67,
  144. 0.67,
  145. 0.67,
  146. 0.67,
  147. 0.67,
  148. 0.67,
  149. 0.67
  150. ]
  151. ]
  152. }
  153. },
  154. "tensorrt": {
  155. "dynamic_shapes": {
  156. "im_shape": [
  157. [
  158. 1,
  159. 2
  160. ],
  161. [
  162. 1,
  163. 2
  164. ],
  165. [
  166. 8,
  167. 2
  168. ]
  169. ],
  170. "image": [
  171. [
  172. 1,
  173. 3,
  174. 640,
  175. 640
  176. ],
  177. [
  178. 1,
  179. 3,
  180. 640,
  181. 640
  182. ],
  183. [
  184. 8,
  185. 3,
  186. 640,
  187. 640
  188. ]
  189. ],
  190. "scale_factor": [
  191. [
  192. 1,
  193. 2
  194. ],
  195. [
  196. 1,
  197. 2
  198. ],
  199. [
  200. 8,
  201. 2
  202. ]
  203. ]
  204. }
  205. }
  206. }
  207. }
  208. }