文件名称:mmdetection-to-tensorrt:将mmdetection模型转换为tensorrt,支持fp16,int8,批输入,动态形状等
文件大小:135KB
文件格式:ZIP
更新时间:2024-04-07 23:52:49
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