Déployer pytorch .la pth modèle dans un script python

0

La question

Après le succès de la formation de mon yolact modèle à l'aide d'un dataset personnalisé je suis heureux avec le résultat retourné par inférence eval.py à l'aide de cette commande à partir d'anaconda terminal:

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

Maintenant, je veux exécuter cette inférence à partir de mon propre script python au lieu de l'utilisation de l'anaconda terminal. Je veux être en mesure d'obtenir les boîtes englobantes des détections effectuées sur webcam images obtenues par le présent code ci-dessous. Une idée ?

import cv2

src = cv2.VideoCapture(0)

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

L'eval.py le code est ici à Yolact référentiel https://github.com/dbolya/yolact/blob/master/eval.py

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

La meilleure réponse

0

Je vais juste écrire le pseudo-code ici pour vous.

Étape 1: Essayez de charger le modèle à l'aide de lignes à partir de et se terminant ici

Étape 2: Utilisez cette fonction pour l'évaluation. Au lieu de cv2.imread, il vous suffit d'envoyer votre image

Étape 3: Suivez cette fonction pour obtenir les boîtes englobantes. Surtout cette ligne. Juste trackback 't' variable et vous obtiendrez vos boîtes englobantes.

Espérons que cela aide. Laissez-moi savoir si vous avez besoin de plus de précisions.

2021-11-20 08:35:20

Suivant thoose je pourrais modifier l'eval.py et obtenir de l'évaluation et de boîtes englobantes. Merci beaucoup pour la réponse, très utile :D
Ramiro Sena

Dans d'autres langues

Cette page est dans d'autres langues

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