inference.yml 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  1. mode: paddle
  2. draw_threshold: 0.5
  3. metric: COCO
  4. use_dynamic_shape: false
  5. Global:
  6. model_name: RT-DETR-L_wireless_table_cell_det
  7. arch: DETR
  8. min_subgraph_size: 3
  9. Preprocess:
  10. - interp: 2
  11. keep_ratio: false
  12. target_size:
  13. - 640
  14. - 640
  15. type: Resize
  16. - mean:
  17. - 0.0
  18. - 0.0
  19. - 0.0
  20. norm_type: none
  21. std:
  22. - 1.0
  23. - 1.0
  24. - 1.0
  25. type: NormalizeImage
  26. - type: Permute
  27. label_list:
  28. - cell
  29. Hpi:
  30. backend_configs:
  31. paddle_infer:
  32. trt_dynamic_shapes: &id001
  33. im_shape:
  34. - - 1
  35. - 2
  36. - - 1
  37. - 2
  38. - - 8
  39. - 2
  40. image:
  41. - - 1
  42. - 3
  43. - 640
  44. - 640
  45. - - 1
  46. - 3
  47. - 640
  48. - 640
  49. - - 8
  50. - 3
  51. - 640
  52. - 640
  53. scale_factor:
  54. - - 1
  55. - 2
  56. - - 1
  57. - 2
  58. - - 8
  59. - 2
  60. trt_dynamic_shape_input_data:
  61. im_shape:
  62. - - 640
  63. - 640
  64. - - 640
  65. - 640
  66. - - 640
  67. - 640
  68. - 640
  69. - 640
  70. - 640
  71. - 640
  72. - 640
  73. - 640
  74. - 640
  75. - 640
  76. - 640
  77. - 640
  78. - 640
  79. - 640
  80. - 640
  81. - 640
  82. scale_factor:
  83. - - 2
  84. - 2
  85. - - 1
  86. - 1
  87. - - 0.67
  88. - 0.67
  89. - 0.67
  90. - 0.67
  91. - 0.67
  92. - 0.67
  93. - 0.67
  94. - 0.67
  95. - 0.67
  96. - 0.67
  97. - 0.67
  98. - 0.67
  99. - 0.67
  100. - 0.67
  101. - 0.67
  102. - 0.67
  103. tensorrt:
  104. dynamic_shapes: *id001