物体検出AIをスマホで使う

Yolov8はポピュラーな物体検出モデルです。
この記事ではYolov８をiOSで使えるようにします。

方法

Yolov8のインストールとモデルチェックポイントのダウンロード。
（モデルのチェックポイント一覧は上記のultralyticsのリポジトリにあります。）

pip install ultralytics
wget https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8s.pt

変換

from ultralytics import YOLO

model = YOLO('yolov8s.pt')  # load an official model
model.export(format='coreml', nms=True)

もし以下のエラーが出た場合は、ultralyticsライブラリのyolo/engine/exporter.pyの以下の行を変更します（390行目）。

# f = self.file.with_suffix('.mlmodel')
# ↓↓↓
f = self.file.with_suffix('.mlpackage')

ptファイルと同じディレクトリに.mlpackageファイル（CoreML）が保存されます。

iOSのカメラで動かすサンプルは以下。

🐣

フリーランスエンジニアです。
AIについて色々記事を書いていますのでよかったらプロフィールを見てみてください。

もし以下のようなご要望をお持ちでしたらお気軽にご相談ください。
AIサービスを開発したい、ビジネスにAIを組み込んで効率化したい、AIを使ったスマホアプリを開発したい、
ARを使ったアプリケーションを作りたい、スマホアプリを作りたいけどどこに相談したらいいかわからない…

いずれも中間コストを省いたリーズナブルな価格でお請けできます。

機械学習やAR技術を使ったアプリケーションを作っています。
機械学習／AR関連の情報を発信しています。