Logo
Explore Help
Register Sign In
mirrors/mmdeploy
1
0
Fork 0
You've already forked mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity
mmdeploy/configs/_base_/backends/tensorrt.py

6 lines
140 B
Python
Raw Normal View History

add tensorrt support (#2)
2021-06-23 13:14:28 +08:00
import tensorrt as trt
add configs, add convert pipeline
2021-06-17 15:28:23 +08:00
backend = 'tensorrt'
changed codebase structure (#10) * 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
2021-07-12 16:26:44 +08:00
tensorrt_params = dict(
shared_params=dict(fp16_mode=False, log_level=trt.Logger.WARNING))
Reference in New Issue Copy Permalink
Powered by Gitea Version: 1.23.8 Page: 2801ms Template: 84ms
English
Bahasa Indonesia Deutsch English Español Français Gaeilge Italiano Latviešu Magyar nyelv Nederlands Polski Português de Portugal Português do Brasil Suomi Svenska Türkçe Čeština Ελληνικά Български Русский Українська فارسی മലയാളം 日本語 简体中文 繁體中文(台灣) 繁體中文(香港) 한국어
Licenses API