तैनात pytorch .pth मॉडल में एक अजगर स्क्रिप्ट

0

सवाल

के बाद सफलतापूर्वक प्रशिक्षण मेरा yolact मॉडल का उपयोग कर एक कस्टम डेटासेट मैं कर रहा हूँ के साथ खुश निष्कर्ष के परिणाम से outputted eval.py इस आदेश का उपयोग करने से एनाकोंडा टर्मिनल:

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

अब मैं चलाने के लिए चाहते हैं इस अनुमान से मेरे खुद के अजगर स्क्रिप्ट का उपयोग करने के बजाय एनाकोंडा टर्मिनल. मैं चाहता हूँ प्राप्त करने में सक्षम हो 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

computer-vision opencv python pytorch
2021-11-19 17:21:59
1

सबसे अच्छा जवाब

0

मैं सिर्फ लिखने के लिए pseudocode यहाँ आप के लिए ।

चरण 1: लोड करने का प्रयास करें मॉडल का उपयोग कर लाइनों से शुरू, यहाँ और यहाँ

चरण 2: उपयोग इस समारोह के लिए मूल्यांकन. के बजाय cv2.imread, तुम बस की जरूरत है भेजने के लिए अपने फ्रेम

कदम 3: का पालन करें, इस समारोह को प्राप्त करने के लिए bounding बॉक्स. विशेष रूप से इस लाइन. बस trackback 'टी' चर और आप मिल जाएगा अपने bounding बॉक्स.

आशा है कि यह मदद करता है । मुझे पता है अगर आप की जरूरत है और अधिक स्पष्टीकरण.

2021-11-20 08:35:20

निम्नलिखित thoose कदम मैं को संशोधित कर सकता है eval.py और मूल्यांकन और bounding बॉक्स. बहुत बहुत धन्यवाद के लिए जवाब है, बहुत उपयोगी है :D
Ramiro Sena

अन्य भाषाओं में

यह पृष्ठ अन्य भाषाओं में है

Русский
..................................................................................................................
Italiano
..................................................................................................................
Polski
..................................................................................................................
Română
..................................................................................................................
한국어
..................................................................................................................
Français
..................................................................................................................
Türk
..................................................................................................................
Česk
..................................................................................................................
Português
..................................................................................................................
ไทย
..................................................................................................................
中文
..................................................................................................................
Español
..................................................................................................................
Slovenský
..................................................................................................................