Wdrożenie modelu pytorch .pth w skrypcie pythona

Po zakończonym mojego modelu yolact z wykorzystaniem własnego zestawu danych jestem zadowolony z wyników wyjściowych pochodzących eval.py za pomocą tego polecenia z terminala 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

Teraz chcę uruchomić ten wniosek z mojego własnego skryptu python zamiast korzystania z terminalu anaconda. Chcę mieć możliwość otrzymywania ograniczające ramy wykryć, wykonanych na klatkach kamery, otrzymanych za pomocą tego kodu poniżej. Jakieś pomysły ?

import cv2

src = cv2.VideoCapture(0)

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

W eval.py kod znajduje się tutaj, w repozytorium Yolact https://github.com/dbolya/yolact/blob/master/eval.py