YOLOによる物体検出

2023年12月版

YOLOによる物体検出をGoogle Colabで動かすコードは次の通りです（公式サイトのサンプルまんまですが）。

ライブラリをインストールする

%pip install ultralytics
import ultralytics
ultralytics.checks()

検出を実行する

!yolo predict model=yolov8n.pt source='https://ultralytics.com/images/zidane.jpg'

検出画像を表示する

from IPython.display import Image,display_jpeg
display_jpeg(Image('runs/detect/predict/zidane.jpg'))

Google Colabのリンクは次の通りです。

Google Colaboratory

＝＝以下は旧バージョンでのコード、現在のGoogle Colabでは動かない＝＝

ライブラリをコピーする

スーパーマケット自体を持ってくる

#ライブラリをコピーする
!git clone https://github.com/ultralytics/yolov5

買ってきた食材YOLO、どこ（キッチン）に入ったことを確認する

インストールしたYOLO

食材の場所に移動する

#場所を移動する
%cd /content/yolov5
%pwd

キッチンのどこに食材を置いたか確認して移動する

YOLOの中に移動する

ライブラリをインストールする

#ライブラリをインストールする
!pip install -r requirements.txt

食材を買い出しにいきましょう

requirements.txtに食材のリストが書かれている。

料理する

#実行してみる
import torch
# Model
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
# Images
img = 'https://ultralytics.com/images/zidane.jpg'
# Inference
results = model(img)
# Results
results.print()
results.show()

美味しい料理が出来上がる

サンプル画像を認識した

Google Driveをマウントする

クリックでもマウントできる

ロッカーにある私物を持ってくるイメージ

アレンジする

場所をコピーする

画像を変更して実行する

物体検出の結果を確認する