可选择从YOLOv8官方主页下载yolov8s-seg.pt模型,或在导出onnx模型中自动下载模型。 安装如下依赖。
pip3 install ultralytics --force-reinstall
pip3 install onnx如果使用tpu-mlir编译模型,则必须先将Pytorch模型导出为onnx模型。YOL0v8官方仓库提供了模型导出接口,可以直接使用它导出onnx模型:
from ultralytics import YOLO
model = YOLO("yolov8s-seg.pt")
model.export(format='onnx', opset=17, dynamic=True)上述脚本会在原始pt模型所在目录下生成导出的yolov8s-seg.onnx等模型。