前回yolov5の導入を紹介したので今回は動作確認の仕方を紹介します。
まずyolov5を入れた場所を確認してください。
anacondaでyolov5を入れるときにインストールする場所を変えてない限りは
C:\Users\ユーザー名\yolov5
に入っているはずです。
yolov5のフォルダ内↓のようになっています。
data/imagesを見ると２枚の画像が入っています
動作確認ではこの画像を使います。
次にAnacondaを起動します。
そしてyolov5に必要なモジュールをインストールした環境を選択して、ターミナルを立ち上げてください。
そしてcd yolov5と入力してディレクトリを変更してください。
python detect.py --source ./data/images/ --weights yolov5s.pt --conf 0.4
とターミナルに入力すると学習されたモデルyolov5s.ptを使ってimagesの中に入っている画像の判定を行います。
結果は
runs/detect/exp
の中に入ります。
２枚目の画像zidane.jpgを見るとpersonとtieで判定されて枠で囲われています。
これでyolov5の動作確認は以上です。
yolov5をGPUで動作させる方法
soiyasoiya58.hatenablog.com
