yolov5のコマンド「python detect.py ……」をターミナルから実行すると、処理の進行に応じて結果が一行ずつ表示されていきます。
Fusing layers...
YOLOv5s summary: 213 layers, 7012822 parameters, 0 gradients
image 1/5 C:*.jpg: 640x480, Done. (0.014s)
image 2/5 C:*.jpg: 640x480, Done. (0.014s)
・
・
image 5/5 C:****************.jpg: 640x480, Done. (0.014s)
Results saved to -----------\yolov5\runs\detect\exp
1 labels saved to ----------\yolov5\runs\detect\exp\labels
やりたいことは、pysimpleguiで作成したOutputウィンド内に、ターミナルと同じように結果をリアルタイムで表示したいです。
下記プログラムを実行したところ、コマンドが完了した段階で以下がOutputに一気に表示されます(リアルタイムで表示されない).
なお、stderrに結果が吐かれることは確認しています。
Fusing layers...
YOLOv5s summary: 213 layers, 7012822 parameters, 0 gradients
image 1/5 C:*.jpg: 640x480, Done. (0.014s)
image 2/5 C:*.jpg: 640x480, Done. (0.014s)
image 3/5 C:*.jpg: 640x480, Done. (0.014s)
image 4/5 C:*.jpg: 640x480, Done. (0.014s)
image 5/5 C:****************.jpg: 640x480, Done. (0.014s)
Results saved to -----------\yolov5\runs\detect\exp
1 labels saved to ----------\yolov5\runs\detect\exp\labels
自分なりに、バッファリングされているのではとflush()などを試しましたがだめでした。
時間のかかる処理なので、コマンドが終了しないと結果がわからないのはうまくないです。
ターミナルから実行した時と同じように、実行結果を逐次(リアルタイムで)表示させる方法についてご教示をよろしくお願いいたします。
(なお、yolov5で提供されている detect.py 自体に手を入れるのは自分ではちょっと難しいです。)
import os,sys
import PySimpleGUI as sg
import subprocess
font=("meiryo", 12)
layout = [
・
全ては省略
・
[sg.Output(size=(120, 30), key='-OUTPUT-')]
]
window = sg.Window('YOLO', layout, text_justification='left', auto_size_text=False)
while True:
event, values = window.read()
if event == sg.WIN_CLOSED:
break
if event == '実 行':
value = sg.popup_ok_cancel('OK/Cancel')
if value == 'OK':
PYTHON = 'C:/Users/ma4ka/anaconda3/envs/py39_yolov5/python.exe -u '
BASEPATH = values['basepath'] + '/detect.py'
SOURCE = ' --source ' + values['imagepath']
WEIGHTS = ' --weights ' + values['weightfile']
LINETHICK = ' --line-thickness ' + str(int(values['linewidth']))
CONFTHRES = ' --conf-thres ' + str(values['confidence'])
SAVETEXT = ' --save-txt'
SAVECONF = ' --save-conf'
command = PYTHON + BASEPATH + SOURCE + WEIGHTS + LINETHICK + CONFTHRES if values['result'] == True: command += SAVETEXT if values['confsave'] == True: command += SAVECONF output_str = subprocess.run(command, capture_output=True, shell=True, text=True) print(output_str.stderr)
window.close()
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2022/05/20 08:21