【文件属性】:
文件名称:mmdetection-to-tensorrt:将mmdetection模型转换为tensorrt,支持fp16,int8,批输入,动态形状等
文件大小:135KB
文件格式:ZIP
更新时间:2021-03-19 06:06:09
inference ssd faster-rcnn object-detection tensorrt
MMDet到张量
该项目旨在将mmdetection模型转换为tensorrt模型end2end。现在专注于对象检测。面膜的支持是实验性的。
支持:
fp16
int8(实验)
批量输入
动态输入形状
不同模块的组合
深度支持
欢迎提供任何建议,错误报告和建议。
执照
该项目是根据。
要求
mmdet> = 2.3.0
重要的!
设置环境变量(在〜/ .bashrc中):
export AMIRSTAN_LIBRARY_PATH= ${amirstan_plugin_root} /build/lib
安装
主持人
git clone https://github.com/grimoire/mmdetection-to-tensorrt.git
cd mmdetection-to-tensorrt
python setup.py develop
码头工人
构建docker镜像(注意Te