Implementar pytorch .la pth en el modelo de una secuencia de comandos de python

0

Pregunta

Tras el éxito de la formación de mi yolact modelo utilizando un conjunto de datos personalizado estoy feliz con la inferencia de resultados emitida por eval.py el uso de este comando de 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

Ahora quiero ejecutar esta inferencia a partir de mi propia secuencia de comandos de python en lugar de utilizar la anaconda terminal. Quiero ser capaz de obtener los recuadros de delimitación de las detecciones realizadas en la webcam fotogramas obtenidos por este código a continuación. Alguna 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

El eval.py el código está aquí en Yolact repositorio https://github.com/dbolya/yolact/blob/master/eval.py

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

Mejor respuesta

0

Me acaba de escribir el pseudocódigo aquí para usted.

Paso 1: Pruebe a cargar el modelo de uso de las líneas a partir de aquí y termina aquí

Paso 2: Utilice esta función para la evaluación. En lugar de cv2.imread, usted sólo necesita enviar su marco

Paso 3: Siga esta función para obtener los recuadros de delimitación. Especialmente en esta línea. Sólo trackback la 't' variable y usted recibirá su bounding boxes.

Espero que ayude. Déjeme saber si usted necesita más aclaraciones.

2021-11-20 08:35:20

Siguiente thoose pasos que se podía modificar el eval.py y obtener la evaluación y las cajas de contorno. Muchas gracias por la respuesta, muy útil :D
Ramiro Sena

En otros idiomas

Esta página está en otros idiomas

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