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
137 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(
[Feature] TensorRT int8 support (#35) * fix custom ops support, fix multiple mark bug, add name mapping * check if the value_info need to be added * remove unnecessary print * add nms implement * two stage split wip * add two stage split * add split retinanet visualize * add two stage split (wip) * finish two stage split * fix lint * move parse string to mmdeploy.utils * add func mark count dict * use assert_cfg_valid * update func count before add Mark * fix dynamic shape support * add calib data generator * create calib dataset * finish end2end int8 * add split two stage tensorrt visualize
2021-08-19 12:56:00 +08:00
shared_params=dict(fp16_mode=False, log_level=trt.Logger.INFO))
Reference in New Issue Copy Permalink
Powered by Gitea Version: 1.23.8 Page: 770ms Template: 5ms
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