yolov5で動画から物体検知できたら画像を保存するシステムの作成

Question

前提・実現したいこと

yolov5を使用して動画から物体検出を行い、検知した際に画像として保存、もしくはスクリーンショットを撮るシステムを作成したい

発生している問題・エラーメッセージ

以下のコードをどこに入れればいいかわかりません。また、コードを追加する必要があるか教えていただければ幸いです。
if num == 0:
        cv2.imwrite("D:\caputure\waitCapture_{}.jpg".format(saveNumber), file_name)
    else:
        cv2.imwrite("D:\caputure\clickCapture_{}.jpg".format(saveNumber), file_name)
    cv2.waitKey(1)

該当のソースコード

試したこと

補足情報（FW/ツールのバージョンなど）

https://github.com/ultralytics/yolov5

Comments

[meg_]

具体的に分からないことは何でしょうか？

[sunred19725]

上記のコードをdetect.pyまたはdatasets.pyのどこに入力すれば良いのか分かりません

[kuma_kuma_]

なぜ質問者様はなぜそのソースコードを入れなければならないと思ったのですか？
GitHubに上がっているソースをそのまま動かして何かエラーが発生したのですか？
その辺りの経緯を詳しく記載下さい。

Answer 1

detect.pyの123行目に追加したらできました