inference.yml 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
  1. mode: paddle
  2. draw_threshold: 0.5
  3. metric: COCO
  4. use_dynamic_shape: false
  5. Global:
  6. model_name: PP-DocLayout_plus-L
  7. arch: DETR
  8. min_subgraph_size: 3
  9. Preprocess:
  10. - interp: 2
  11. keep_ratio: false
  12. target_size:
  13. - 800
  14. - 800
  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. - paragraph_title
  29. - image
  30. - text
  31. - number
  32. - abstract
  33. - content
  34. - figure_title
  35. - formula
  36. - table
  37. - reference
  38. - doc_title
  39. - footnote
  40. - header
  41. - algorithm
  42. - footer
  43. - seal
  44. - chart
  45. - formula_number
  46. - aside_text
  47. - reference_content
  48. Hpi:
  49. backend_configs:
  50. paddle_infer:
  51. trt_dynamic_shapes: &id001
  52. im_shape:
  53. - - 1
  54. - 2
  55. - - 1
  56. - 2
  57. - - 8
  58. - 2
  59. image:
  60. - - 1
  61. - 3
  62. - 800
  63. - 800
  64. - - 1
  65. - 3
  66. - 800
  67. - 800
  68. - - 8
  69. - 3
  70. - 800
  71. - 800
  72. scale_factor:
  73. - - 1
  74. - 2
  75. - - 1
  76. - 2
  77. - - 8
  78. - 2
  79. trt_dynamic_shape_input_data:
  80. im_shape:
  81. - - 800
  82. - 800
  83. - - 800
  84. - 800
  85. - - 800
  86. - 800
  87. - 800
  88. - 800
  89. - 800
  90. - 800
  91. - 800
  92. - 800
  93. - 800
  94. - 800
  95. - 800
  96. - 800
  97. - 800
  98. - 800
  99. - 800
  100. - 800
  101. scale_factor:
  102. - - 2
  103. - 2
  104. - - 1
  105. - 1
  106. - - 0.67
  107. - 0.67
  108. - 0.67
  109. - 0.67
  110. - 0.67
  111. - 0.67
  112. - 0.67
  113. - 0.67
  114. - 0.67
  115. - 0.67
  116. - 0.67
  117. - 0.67
  118. - 0.67
  119. - 0.67
  120. - 0.67
  121. - 0.67
  122. tensorrt:
  123. dynamic_shapes: *id001