はじめに
今回するのは、物体検出手法であるYOLO11の既存モデルの実装です。
簡単にYOLOについて紹介しますが、詳しくはこちらを見ていただけるとわかりやすいかと思います。
YOLO:物体検出手法の一種であり、動画像内の物体に対して物体名と信頼度の表示がリアルタイムで可能。
YOLO11:YOLOのバージョン11であり、検出精度と速度が上昇している。「物体検出」に加えて、「インスタンス分類」、「画像分類」、「姿勢推定」、「指向性物体検出」が利用可能。
既存モデル:事前に用意されている学習済みのモデルであり、犬や猫など80種類の物体を学習なしで検出可能。
->以下、物体検出の例です。 スーツや顔が検出されていないのは、既存モデルに含まれていないからです。
物体検出の実装
本題の実装です。実装方法は色々ありますが、Googleアカウントさえあれば、基本的に無料で出来るGoogle Colaboratoryで「物体検出」をやっていきます。
①Google Colaboratoryを開く
ドライブを開き、新規を押下すると、以下のようなメニューが表示されるので、「その他」からGoogle Colaboratoryが開けます。ない場合は、アプリを追加を押下し、「colab」で検索すると出てくると思うので、インストールします。
Google Colaboratory開くと、以下のような画面が出てくるので、①は完了です。
②Google Colaboratoryの基本的な操作
小さくて恐縮ですが、以下の赤・緑・黄の枠を付けた機能が大事なので、それぞれ説明します。
赤枠:繋がないと何もできないので、「接続」を押下し、ランタイムに繋ぎます。繋ぐと、チェックマークやら出てきます。
緑枠:コードを打ち込み、再生ボタンのようなボタンを押すことで、実行できます。
黄枠:「挿入」を押下し、「コードセル」を押下する事で、コード打つ所を足せます。
③Google driveのマウント
以下のコードを打ち込み実行し、アカウント選択やアクセス許可することで、自分のドライブにあるフォルダやファイルが使えます。チェックマークが付いてれば、成功です。
「 from google.colab import drive 」
「 drive.mount('/content/drive') 」
④ultralyticsのインストール
以下のコードを打ち込むことで、ultralyticsライブラリを Python 環境にインストールします。私はalready satisfiedと出力されていますが、初回だと色々インストールされてもっと長いです。
「 !pip3 install ultralytics 」
⑤YOLO11のインポート
以下のコードを打ち込むことで、先ほどインストールしたultralyticsを利用することができます。また、modelを指定することで、最新のYOLO11が利用できます。これも私は初回ではないので何も出力されませんが、初回の方は出力されます。
「 from ultralytics import YOLO 」
「 model = YOLO("yolo11n.pt") 」
⑥既存モデルでの物体検出
以下のコードを参考に打ち込むことで、YOLO11が勝手に物体検出してくれます。
- 「”/content/drive/MyDrive/try/cat.jpg”」:検出したい画像のパス
- 「save = True」:検出した画像を保存
- 「project=”/content/drive/MyDrive”」(任意):保存したいフォルダを指定
「 results = model("検出したい画像のパス", save=True, project="保存したいフォルダへのパス") 」
↑の画像の出力を見ると、「Results saved to /content/drive/MyDrive/predict」とありますが、ここに出力された画像があります。
画面左側にフォルダのマークがあるのでそれを押すと、↓の画像のような画面が出てくるので、パス通り進んでください。
ちなみにですが、検出したい画像を指定する際は、画像の上で右クリックすると、パスをコピーと出てくるので、コピペすると早いです。
predictフォルダの下にあるcatという画像が今回出力された画像で、それを押下すると以下のように表示されます。
結果を見ると、catの確率が87%と出力されているので、検出できていますね。
以上がYOLO11での既存モデルを用いた物体検出です。意外と簡単にできるので、やってみてください。
参考文献
iret.media: 【物体検出手法】YOLO11とは?,(2025/3/7)
ultralytics: COCO データセット, (2025/3/7)
Google: Colaboratory へようこそ,(2025/3/7)
ultralytics: Ultralytics YOLO11,(2025/3/7)
画像提供(猫):同期のHくん