YOLOv5で検出結果の出力を保存する方法（Google Colaboratory）

1535 ワード
YOLOV5 GoogleColaboratory GoogleColaboratory テキストリンク

はじめての投稿です。

該当の方法についてはGoogle Colabratoryでの話なので他の環境について把握していませんのであしからず。

先に結論。

! python detect.pyへのパス --source 動画ファイルへのパス 2> XXXXX.txt

例えば自分の場合、

! python /content/yolov5/detect.py --source /content/drive/MyDrive/otaku.mp4 2> list.txt

といった感じ。
ちなみに出力結果のファイル名の前に保存したいディレクトリのパスを記載すれば任意の場所にほぞんできます。

参考記事として、

があります。

もともと、上記の二番目のリンク先と同じことがやりたくてdetect.pyで出力されるフレームごとの検出結果を.txtでも.csvでもなんでもいいから保存できないかと考えていました。

参考記事の通り>を入れてもなぜか上手くいかず、かなりの時間を要しました

どうやら該当の処理のことをlinuxでのリダイレクトというものに該当するとのことで、またリダイレクトにも種類があるということがわかりました。(詳細は調べてください）

参考になれば幸いです。