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 Colabでは動かない==
ライブラリをコピーする
#ライブラリをコピーする
!git clone https://github.com/ultralytics/yolov5
買ってきた食材YOLO、どこ(キッチン)に入ったことを確認する
食材の場所に移動する
#場所を移動する
%cd /content/yolov5
%pwd
ライブラリをインストールする
#ライブラリをインストールする
!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をマウントする
アレンジする
この記事が気に入ったらサポートをしてみませんか?