LoginSignup
2
3

More than 1 year has passed since last update.

自前データで学習したyolov5モデルを使ってgoogle colab上で物体検出

Posted at

前回の記事の補足になります。独自データで訓練させたyolov5モデルをgoogle colab上で動かす、動作確認の方法です。
↓前回の記事
独自データでのGooglebColabでのyolov5モデルの訓練とローカルでの実行

独自yolov5モデルをcolab上で動かすので、環境構築は不要ですがwebcamからのリアルタイム検出には未対応です。

訓練でできたモデルxxxxpt.ファイルをダウンロード

ダウンロードフォルダに移動して名前の変更.

last.pt → 20220815_last.pt など自分でわかりやすいものに変更
(全角は使わない、スペースは使わない)

githubに移動

My_github
2022-08-15.png

左上のOpen in Colabからgoogle colabに移動(google アカウント必要)

 最初の2つのセルを実行して、動作確認(ジダンが検出されればOK)
2022-08-15 (1).png

重み付きptファイルのアップロード

 yolov5フォルダ直下に20220815_last.ptをアップロード
(ファイルを左下スミにドロップするだけ、アップロードしたファイルはgoogle colabの接続が切れると削除される)

2022-08-15 (2).png

検出させたい画像(動画)のアップロード

フォルダ yolov5/data/images に検出させたい動画をアップロード
あとで動画ファイル名はシンプルが無難

2022-08-15 (3).png

detect.py セルで--source --weightの書き換えと実行

!python detect.py --source /content/yolov5/data/images/ocd.MP4 
--weights /content/yolov5/20220815_last.pt 
--conf 0.3

これを書き換える↓

!python detect.py --source /content/yolov5/data/images/アップロードしたファイル名.MP4 
--weights /content/yolov5/アップロードした訓練済みファイル名.pt 
--conf 0.3

セルを実行 検出が開始される!
2022-08-15 (4).png

yolov5/runs/detect/exp xx (←xxには複数回検出を行った場合xx回目の数字が入る)
にmp4ファイルが作成されているので(動画の場合)これを右クリックでダウンロード
2022-08-15 (5).png

 

2
3
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
2
3