Yolov3(もしくはYolov5)をPythonのソースコード内で呼び出す方法

Question

前提

閲覧して頂きありがとうございます．
現在大学の研究においてOpenCVを用いて動画に対してフレームカット処理を行い，それらの画像に対して物体検出(主に人)を行おうとしています．

実現したいこと

以下の画像のようにターミナルから実行を行い検出を行える環境は構築いたしました．
よってこの結果をpythonのソースコード内から呼び出して出力結果として同じものを出力したいと思っています．
ファイルのimportなどの部分が関連してくると思っているのですがファイルの呼び出し等の知識が乏しく，ご教示いただけると幸いです．

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

python 3.10.4
Mac
pytorch_Yolov3

Comments

[kazto]

> 以下の画像のようにターミナルから実行を行い検出を行える環境は構築いたしました．
どのような手順で何をやったかを示していただかないと、アドバイスのしようがないかと思います。

[arapon516]

ご指摘ありがとうございます，基本的にはこちらのサイトの方を参照させて頂いて実装させていただきました．https://pystyle.info/pytorch-yolov3-how-to-use-pretrained-model/　そして該当サイトに書かれているコード等も試してみたのですがImport "yolov3.detector" could not be resolvedとなり，呼び出せない状況です．

[jbpb0]

> Import "yolov3.detector" could not be resolved

https://pystyle.info/pytorch-yolov3-how-to-use-pretrained-model/
の「画像から人を検出する」の
「yolov3_path に git clone した pytorch_yolov3 ディレクトリのパスを指定してください。」
がちゃんとできてたら、
https://github.com/nekobean/pytorch_yolov3/blob/main/yolov3/detector.py
がインポートできるはず

sys.path.append(str(yolov3_path))

のすぐ下に、

import sys
import pprint
pprint.pprint(sys.path)

を追加して実行したら、「sys.path」に登録されてるパスが表示されますので、それを見てpytorch_yolov3のパスが正しく設定されてるかを確認してみてください

「yolov3_path = Path("...」で指定するパスは、絶対パスの方がいいかも

[arapon516]

絶対パスにすることでうまく行きました．ありがとうございます．．．．！！
お二方ともご丁寧に対応してくださってありがとうございました！！
YOLOv5でもできるように頑張ります．．．！

Answer 1

Import "yolov3.detector" could not be resolved

YOLOv3 – 学習済みモデルで画像から人や車を検出する方法
の「画像から人を検出する」の
「yolov3_path に git clone した pytorch_yolov3 ディレクトリのパスを指定してください。」
がちゃんとできてたら、
detector.py
がインポートできるはず

　

sys.path.append(str(yolov3_path))

のすぐ下に、

import sys
import pprint
pprint.pprint(sys.path)

を追加して実行したら、「sys.path」に登録されてるパスが表示されますので、それを見てpytorch_yolov3のパスが正しく設定されてるかを確認してみてください

　
「yolov3_path = Path("...」で指定するパスは、絶対パスの方がいいかも

Answer 2

現状どのような環境構築しているか不明ですがyolov5を利用するとよいのではないかと思います。
Pythonでのサンプルコードもあります。