После успешного обучения моей модели yolact с использованием пользовательского набора данных я доволен результатами вывода, полученными eval.py используя эту команду из терминала anaconda:
python eval.py --trained_model=./weights/yolact_plus_resnet50_abrasion_39_10000.pth --config=yolact_resnet_abrasion_config --score_threshold=0.8 --top_k=15 --images=./images:output_images
Теперь я хочу запустить этот вывод из моего собственного скрипта python вместо использования терминала anaconda. Я хочу иметь возможность получать ограничительные рамки обнаружений, сделанных на кадрах веб-камеры, полученных с помощью этого кода ниже. Есть идеи ?
import cv2
src = cv2.VideoCapture(0)
while True:
ret, frame = src.read()
cv2.imshow('frame', frame)
key = cv2.waitKey(5)
if key == (27):
break
В eval.py код находится здесь, в репозитории Yolact https://github.com/dbolya/yolact/blob/master/eval.py