跳转到内容
新建笔记

tensorrt:YOLOV8 用 C++实现分割任务的engine导出说明

在实现过程中遇到情况说明:

1. 通过命令:

yolo export model=best.pt format=engine

导出的best.engine在用C++部署时会出现不兼容的情况:

Error Code 1: Serialization (Serialization assertion magicTagRead == kMAGIC_TAG failed.Magic tag does not match)

其中,python的tensorrt版本与C++的tensorrt版本一致。

2. 通过命令:

yolo export model=best.pt format=onnx
python -m onnxsim best.onnx best_sim.onnx
trtexec trtexec --onnx=best_sim.onnx --saveEngine=best_sim.engine --fp16

导出的engine,使用C++部署无类似报错。


来源:CSDN 原文,首次发布于 2024-06-22。