Yolov7の組み込み方がわかりません

前提

Pythonで画面をmssで読み取り、それをYolov7で画像解析し、その結果を、imshowで表示するプログラムを作っています。
しかしYolov7の画像解析がプログラムに組み込めなくて困っています。

実現したいこと

プログラム内にYolov7を組み込む。

途中までのプログラム

import sys
from pathlib import Path
import cv2
import numpy
import mss

with mss.mss() as sct:
    monitor = (190,26,648,796)

#学習モデルのパス
weight = "~~~~~~~~~~~~"


while True:
    try:
        img = numpy.array(sct.grab(monitor))

ここにimgをYolov7で画像解析するプログラムを書く

        cv2.imshow("image", img)


        # Press "q" to quit
        if cv2.waitKey(25) & 0xFF == ord("q"):
            cv2.destroyAllWindows()
            break
    
    except:
        continue

実行している環境

OS:MacOS monterey

プログラム初心者です。お願いします。

退会済みユーザー

2022/12/22 02:17 編集

https://github.com/WongKinYiu/yolov7/blob/main/detect.py#L88 ここのimgがTorchでのimg（0-1で正規化）されたものが突っ込まれています。あとはpredに推論結果（ラベル、信頼度、座標）が入ってくるはずなので、そのラベルらの情報を適当にcv2.rectangleやcv2.putTextでnumpyのimgに乗っければ良さそうです。ただ、ここで掲載されているコードよりもYOLO7のコードの方が圧倒的に多いと思いますので、自作のコードにYOLO7を付け加えるより、YOLO7のコードを生かしながら自分のやりたい処理をデバグしながら（エラーが起きないように、やりたい方向に徐々に近づいているか変数を少しづずprint()しながら）ねじ込んでいく方がうまくいくはずです。このアプローチは初心者だろうが慣れている人だろうが、YOLO7を使ったことがない人であれば基本的に変わりません。その苦労を自分で背負おうとしていれば助けてもらえますし、だれか書いてくれないかな、だと「やってほしいことを丸投げした質問」となって回答がつかなくなります。あとは12510196さんができないなりにやれるところまでやるだけです。