今回はうちのネコの写真を使って物体検出をしてみました。
方法は、Google Colab で「YOLO v5」を使用すといった感じです。
Google Colab
Google Colab は以下のサイトから使用します。
colab.research.google.com
YOLO v5 を Google Colab にインストール
google colab で以下のコマンドを実行します。
!git clone https://github.com/ultralytics/yolov5
clone 後に必要なライブラいをインストールします。
%cd /content/yolov5/ !pip install -qr requirements.txt
物体検出
GPU を使用する設定
Google Colab は無料で GPU を使えるので設定しておきます。
「ランタイム>ランタイムのタイプを変更」から「ハードウェア アクセラレータ」を「GPU」に変更します。
物体検出
写真をアップロードして、以下のコマンドで物体検出を行います。
!python detect.py --source ../imgs/cat_01.png --weights yolov5s.pt --conf 0.3 --name demo --exist-ok
ちゃんと検出できてる。
まさかの犬判定!