上传者: 42175516
|
上传时间: 2022-02-19 23:27:54
|
文件大小: 135KB
|
文件类型: -
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