Enable BatchedNMS Plugin in TensorRT #934

jiangjiajun · 2022-10-26T11:19:39Z

由于Mod不被TensorRT支持，当输入为INT类型时，elementwise_mod的转换逻辑改为

tmp0 = x // y
tmp1 = y * tmp0
result = x - tmp1

* Export dynamic dimenstion with string * fix ci error * Update test_quantize_model_speedup.py

jiangjiajun added 7 commits October 26, 2022 11:18

Modify conversion method for elementwise_mod with tensorrt backend

3e7b152

fix bug

cecc8e5

add efficient_nms plugin

2dfb075

Update multiclass_nms.cc

65a6412

Export dynamic dimension with string (#931) (#940)

e16719f

* Export dynamic dimenstion with string * fix ci error * Update test_quantize_model_speedup.py

Fix compile problem

d0f2631

fix error

f0e2e44

jiangjiajun changed the title ~~Modify conversion method for elementwise_mod with tensorrt backend~~ Enable BatchedNMS Plugin in TensorRT Nov 2, 2022

Export dynamic dimension with string (#931) (#944)

80a8c6e

* Export dynamic dimenstion with string * fix ci error * Update test_quantize_model_speedup.py

jiangjiajun merged commit c620954 into develop Nov 2, 2022

jiangjiajun deleted the mod branch November 2, 2022 07:31

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Enable BatchedNMS Plugin in TensorRT #934

Enable BatchedNMS Plugin in TensorRT #934

jiangjiajun commented Oct 26, 2022 •

edited

Enable BatchedNMS Plugin in TensorRT #934

Enable BatchedNMS Plugin in TensorRT #934

Conversation

jiangjiajun commented Oct 26, 2022 • edited

jiangjiajun commented Oct 26, 2022 •

edited