Distribuire pytorch .pth modello in uno script python

0

Domanda

Dopo il successo di allenamento il mio yolact modello utilizzando un set di dati personalizzato sono felice con l'inferenza risultati emessi dal eval.py utilizzando questo comando da anaconda terminale:

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

Ora voglio eseguire questa deduzione dal mio script python invece di utilizzare l'anaconda terminale. Voglio essere in grado di ottenere i riquadri dei rilevamenti effettuati in webcam fotogrammi ottenuti da questo codice riportato di seguito. Qualche idea ?

import cv2

src = cv2.VideoCapture(0)

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

Il eval.py il codice è qui a Yolact repository https://github.com/dbolya/yolact/blob/master/eval.py

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

Migliore risposta

0

Mi limito a scrivere lo pseudocodice qui per voi.

Passo 1: caricare il modello utilizzando le linee a partire da qui e termina qui

Passaggio 2: Utilizzare questa funzione per la valutazione. Invece di cv2.imread, basta inviare il tuo telaio

Passo 3: Seguire questa funzione per ottenere i riquadri di delimitazione. In particolare questa linea. Solo il trackback a 't' variabile e avrete il vostro riquadri di delimitazione.

Speranza che aiuta. Fatemi sapere se avete bisogno di ulteriori chiarimenti.

2021-11-20 08:35:20

Seguenti thoose passi ho potuto modificare il eval.py e ottenere la valutazione e riquadri. La ringrazio molto per la risposta, molto utile :D
Ramiro Sena

In altre lingue

Questa pagina è in altre lingue

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