Après le succès de la formation de mon yolact modèle à l'aide d'un dataset personnalisé je suis heureux avec le résultat retourné par inférence eval.py à l'aide de cette commande à partir d'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
Maintenant, je veux exécuter cette inférence à partir de mon propre script python au lieu de l'utilisation de l'anaconda terminal. Je veux être en mesure d'obtenir les boîtes englobantes des détections effectuées sur webcam images obtenues par le présent code ci-dessous. Une idée ?
import cv2
src = cv2.VideoCapture(0)
while True:
ret, frame = src.read()
cv2.imshow('frame', frame)
key = cv2.waitKey(5)
if key == (27):
break
L'eval.py le code est ici à Yolact référentiel https://github.com/dbolya/yolact/blob/master/eval.py