YOLOv10で鳥を見つけよう！Ultralyticsを使用した物体検出

はるか

今日はYOLOv10について動かしてみよう。

ふゅか

YOLOv10は最新の物体検出モデルで、リアルタイムでの精度がすごく高いんだよ！

モデルの概要

YOLOv10は、物体検出モデルで、NMS（非最大抑制）を排除し、効率と精度を高める新しいアーキテクチャを採用しています。このモデルは、画像内の物体をリアルタイムで正確に検出することを目指しており、従来のYOLOシリーズの欠点を克服しています。特徴抽出ではCSPNetを採用。

モデルの種類

YOLOv10-N, S, M, B, L, Xの各バージョンが存在し、様々な用途に対応。YOLOv10-Xが最も性能が良く、YOLOv10-Nはリソースが限られたときに使用することを想定している。

はるか

YOLOv10-Xは高性能で、YOLOv10-Nはリソースが限られている環境向け。

ふゅか

性能と環境を考慮して適切なモデルを選ぶ必要があるわね！

物体検知

ふゅか

image.pngという画像を用意して物体検知を行ってみよう！

環境

ライブラリ

次のコマンドでライブラリをインストールします。

pip install ultralytics

pythonコード

from ultralytics import YOLO

# モデルのロード
model = YOLO("yolov10x.pt")

# 画像の物体検出
results = model("image.jpg")
results[0].show()

• ライブラリのインポート:
  • ultralyticsライブラリからYOLOクラスをインポート。
• モデルのロード:
  • YOLOv10モデルをロード。
  • モデルの重みファイルを指定。モデルの重みは次のように選択可能。

• 画像の物体検出:
  • 指定した画像ファイルに対して物体検出を実行。
  • 検出結果を変数に格納。
• 結果の表示:
  • 検出結果のうち最初の結果を表示。

実行結果

• すべての結果でbirdであると判定
• YOLOv10-Nだけ体全体をバウンディングボックスで囲むことができなかった。（しっぽの部分）
• YOLOv10-Nが最も確率が低く判定(89%)された。
• YOLOv10-X,Bが最も確率が高く判定(96%)された。

はるか

今回使用した画像の場合はYOLOv10-XとBが最も高い精度。

ほかのプログラム

results[0].speed

で実行にかっかたそれぞれの処理の時間を確認することができます。

{'preprocess': 1.1472702026367188,
 'inference': 73.80056381225586,
 'postprocess': 14.999866485595703}

物体検出に関連する書籍

物体検出とGAN、オートエンコーダー、画像処理入門 PyTorch/TensorFlow2による発展的・実装ディープラーニング/チーム・カルポ

posted with カエレバ

bookfanプレミアム

Yahooショッピングで探す

Amazonで探す

楽天市場で探す

画像認識機械学習プロフェッショナルシリーズ

posted with カエレバ

WINDY BOOKS on line

Yahooショッピングで探す

Amazonで探す

楽天市場で探す

今すぐ試したい!機械学習・深層学習〈ディープラーニング〉画像認識プログラミングレシピ/川島賢

posted with カエレバ

bookfanプレミアム

Yahooショッピングで探す

Amazonで探す

楽天市場で探す