Nasadiť pytorch .pth model v python skript

0

Otázka

Po úspešnom tréningu môj yolact modelu pomocou vlastnej dataset som spokojný s záver výsledky výstup podľa eval.py pomocou tohto príkazu z anaconda terminálu:

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 chcem spustiť tento záver z mojej vlastnej python skript namiesto použitia anaconda terminálu. Chcem byť schopní získať ohraničenie políčka zistenia vykonané na kameru rámy získané tento kód nižšie. Akýkoľvek nápad ?

import cv2

src = cv2.VideoCapture(0)

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

Na eval.py kód je tu na Yolact úloţiska https://github.com/dbolya/yolact/blob/master/eval.py

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

Najlepšiu odpoveď

0

Ja sa len napísať pseudocode tu pre vás.

Krok č. 1: Skúste modelu pomocou riadky začínajúce od tu a končí tu

Krok 2: Použitie tejto funkcie hodnotenia. Namiesto cv2.imread, stačí poslať vašu rám

Krok 3: Sledujte túto funkciu, ak chcete získať ohraničenie polí. Zvlášť v tomto riadku. Len trackback 't' premenná a dostanete váš ohraničenie polí.

Dúfam, že to pomáha. Dajte mi vedieť, ak budete potrebovať ďalšie objasnenie.

2021-11-20 08:35:20

Nasledujúce thoose kroky, ktoré by som mohol zmeniť eval.py a získajte hodnotenia a ohraničenie polí. Ďakujem vám veľmi pekne za odpoveď, veľmi užitočné :D
Ramiro Sena

V iných jazykoch

Táto stránka je v iných jazykoch

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