yolov7导出可被trt解析的onnx

(1)注意要使用分支u5版(不需要任何修改直接导出即可)的进行onnx导出(导出后再进行simpify可以进一步消除导出onnx中存在的unsqueeze、gather、shape算子)

(2)如果使用master中的导出onnx,要获得cat三个检测层的结果,要传入--grid

而且发现onnxsim会报错。大老师已经可以成功简化,但是依然存在scatternd算子,这个算子存在就意味着要使用这个插件(虽然高版本的trt已经有了,但是依然要加载官方插件,还是很烦的!)

所以为了避免这个scatternd算子,可以修改导出代码,具体修改后面再搞。

c++部署代码如下:

GitHub - hewen0901/yolov7_trt: yolov7目标检测算法的c++ tensorrt部署代码

还有yolou、杜老的都可以、GitHub - Linaom1214/TensorRT-For-YOLO-Series: tensorrt for yolo series, nms plugin support

你可能感兴趣的:(深度学习算法部署,ONNX,深度学习,部署,深度学习)