เตรียมที่จะออกจะแท๊ pytorch.pth รุ่นในสคริปต์เป็นภาษาไพธอน

หลังจากเรียบร้อยแล้การฝึกของฉัน yolact รุ่นโดยใช้กำหนดวันที่ฉันมีความสุขกับ inference ผล outputted ย 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

ตอนนี้ฉันต้องการที่จะสั่งให้ inference จากของตัวเองเป็นภาษาไพธอนสคริปต์แทนที่จะใช้ anaconda เทอร์มินัล. ผมต้องการที่จะสามารถเอา bounding กล่องของ detections ทำบนเว็บแคมเฟรมได้รับมาจากการยื่นโดยรหัสทางด้านล่างนี้ รู้บ้างมั้ย?

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