Implementa pytorch .pth model într-un script python

După formarea cu succes meu yolact model folosind un custom set de date sunt fericit cu concluzia rezultatele scoase de eval.py folosind această comandă de la 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

Acum vreau pentru a rula această inferență de la propriul meu script python în loc de a folosi anaconda terminal. Vreau să fie capabil de a obține de încadrare cutii de detectii făcut pe webcam cadre obținute prin acest cod de mai jos. Vreo idee ?

import cv2

src = cv2.VideoCapture(0)

while True:
    ret, frame = src.read()
    cv2.imshow('frame', frame)
    key = cv2.waitKey(5)
    if key == (27):
        break

La eval.py codul este aici, la Yolact depozit https://github.com/dbolya/yolact/blob/master/eval.py