mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
9f0e2d4339
* add delta2bbox rewriter * rename onnx2trt * add rewriter: anchor_generator_single_level_grid_priors * Revert "add rewriter: anchor_generator_single_level_grid_priors" This reverts commit ac7cf272942c4787bf143c0d67e414b0d2603b75. * update comments * remove clamp rewriter * remove unused func * reconstruct codes * remomve mark * create export package for codebases * remove grid_sampler symbolic * reorganize nms parts * fix conflicts * update unittest
6 lines
140 B
Python
6 lines
140 B
Python
import tensorrt as trt
|
|
|
|
backend = 'tensorrt'
|
|
tensorrt_params = dict(
|
|
shared_params=dict(fp16_mode=False, log_level=trt.Logger.WARNING))
|