基于TensorRT的yolov5 实例分割部署

yolov5-7.0 github: https://github.com/ultralytics/yolov5/tree/master

1. 代码的使用

1.1 训练yolov5-seg模型

使用的yolov5-7.0的代码，github下载：https://github.com/ultralytics/yolov5/releases/tag/v7.0

训练指令

python segment/train.py --data coco128-seg.yaml --weights yolov5s-seg.pt --cfg yolov5s-seg.yaml

1.2 生成TensorRT engine文件

可以用yolov5/v6.2的官方代码直接导出engine模型，但导出的模型只能在你当前的电脑上使用。建议先导出onnx模型，然后再使用我提供的代码。这样做的好处是，即使你改变了计算机的配置࿰