ものづくりのブログ

うちのネコを題材にしたものづくりができたらいいなと思っていろいろ奮闘してます。

うちのネコを検出してみる①

今回はうちのネコの写真を使って物体検出をしてみました。
方法は、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


ちゃんと検出できてる。
f:id:a1026302:20220305110249p:plain

まさかの犬判定!
f:id:a1026302:20220305110343p:plain