用OpenVINO C++ API編寫YOLOv8-Seg實例分割模型推論程式

Posted By 戰鵬州 on 5 月 26, 2023 in OpenVINO專欄, OpenVINO教學文, 教學文 | 0 comments

1.1 簡介

本文章將介紹使用OpenVINO C++ API開發YOLOv8-Seg實例分割(Instance Segmentation)模型的AI推論程式。本文C++範例程式的開發環境是Windows + Visual Studio Community 2022，請讀者先配置基於Visual Studio的OpenVINO C++開發環境。

請複製本文的程式碼倉：git clone https://gitee.com/ppov-nuc/yolov8_openvino_cpp.git

1.2 匯出YOLOv8-Seg OpenVINO IR 模型

YOLOv8是Ultralytics公司基於YOLO框架，發佈的一款針對物體檢測與追蹤、實例分割、影像分類和姿態估計任務的SOTA模型工具套件。

首先用命令pip install -r requirements.txt 安裝ultralytics和openvino-dev。

然後使用命令：yolo export model=yolov8n-seg.pt format=onnx，完成yolov8n-seg.onnx模型匯出，如下圖所示。

接著使用命令：mo -m yolov8n-seg.onnx --compress_to_fp16，

最佳化並匯出FP16精度的OpenVINO IR格式模型，如下圖所示。