Tras el éxito de la formación de mi yolact modelo utilizando un conjunto de datos personalizado estoy feliz con la inferencia de resultados emitida por eval.py el uso de este comando de anaconda terminal:
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
Ahora quiero ejecutar esta inferencia a partir de mi propia secuencia de comandos de python en lugar de utilizar la anaconda terminal. Quiero ser capaz de obtener los recuadros de delimitación de las detecciones realizadas en la webcam fotogramas obtenidos por este código a continuación. Alguna idea ?
import cv2
src = cv2.VideoCapture(0)
while True:
ret, frame = src.read()
cv2.imshow('frame', frame)
key = cv2.waitKey(5)
if key == (27):
break
El eval.py el código está aquí en Yolact repositorio https://github.com/dbolya/yolact/blob/master/eval.py