YOLOv7で画像にオブジェクトカウンターをつける方法をメモしておきます。
T. Nukuiさんの記事を参考にしました。
YOLOv7では、物体検出を行うことができます。
従来のコードでは、画像にオブジェクトカウンターをつけることはできません。
そこで、画像にオブジェクトカウンターをつけることができるコードを調べたところ、T. Nukuiさんの記事を見つけました。
T. Nukuiさんの記事ではmuratali016さんのコードが使われており、私もmuratali016さんのコードを使います。
従来のコード
Google Colaboratoryにて実行しました。
ランタイム->ランタイムのタイプを変更->T4 GPUを選択
from google.colab import drive
drive.mount('/content/drive')
%cd "/content/drive/MyDrive"
!git clone https://github.com/WongKinYiu/yolov7
%cd "/content/drive/MyDrive/yolov7"
!pip install -r requirements.txt
バウンディングボックスの色を指定するためにdetect.pyの62行目に以下のコードを挿入します。
detect.py
# 62行目
colors[0] = (204, 122, 122)
colors[27] = (255, 102, 102)
物体検出（結果はyolov7/runs/detectに保存される）
%cd "/content/drive/MyDrive/yolov7"
!python detect.py --weights yolov7.pt --conf 0.3 --img-size 640 --source "/content/drive/MyDrive/yolov7/inference/images/zidane.jpg"
この画像が出力されます。
オブジェクトカウンターはついていないことがわかります。
画像にオブジェクトカウンターをつける
detect_and_count.pyをyolov7ディレクトリに配置します。
バウンディングボックスの色を指定するためにdetect_and_count.pyの79行目に以下のコードを挿入します。
detect_and_count.py
# 79行目
colors[0] = (204, 122, 122)
colors[27] = (255, 102, 102)
そして、以下のコードを実行します。
%cd "/content/drive/MyDrive/yolov7"
!python detect_and_count.py --weights yolov7.pt --conf 0.3 --img-size 640 --source "/content/drive/MyDrive/yolov7/inference/images/zidane.jpg"
右下に黄色のオブジェクトカウンターがつきました。
カウンターの色を白色に変えるため、detect_and_count.pyの29行目を以下のように書き換えます。
detect_and_count.py
# 29行目
cv2.putText(im0, str(a) ,(int(align_right),align_bottom), cv2.FONT_HERSHEY_SIMPLEX, 1,(255,255,255),1,cv2.LINE_AA) # colorを(255,255,255)に変更
以下のコードをもう一回実行します。
%cd "/content/drive/MyDrive/yolov7"
!python detect_and_count.py --weights yolov7.pt --conf 0.3 --img-size 640 --source "/content/drive/MyDrive/yolov7/inference/images/zidane.jpg"
カウンターの色が白色になりました。
muratali016さんのコードにより、画像にオブジェクトカウンターをつけることができました。
また、コードを書き換えることにより、カウンターの色を変えることができました。