Implantar pytorch .pth modelo em um script python

0

Pergunta

Depois do sucesso com o treinamento da minha yolact modelo usando um dataset personalizada estou feliz com a inferência de resultados produzidos pelo eval.py usando este comando a partir 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

Agora eu quero para executar esta inferência a partir do meu próprio script python em vez de usar o anaconda terminal. Eu quero ser capaz de obter as caixas delimitadoras de detecções feitas na webcam quadros obtidos por este código abaixo. Alguma idéia ?

import cv2

src = cv2.VideoCapture(0)

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

O eval.py o código está aqui no repositório Yolact https://github.com/dbolya/yolact/blob/master/eval.py

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

Melhor resposta

0

Vou apenas escrever o pseudocódigo aqui para você.

Passo 1: Tente carregar o modelo utilizando as linhas de partida a partir do aqui e termina aqui

Passo 2: Utilize esta função para avaliação. Em vez de cv2.imread, você só precisa enviar o seu quadro de

Passo 3: Execute esta função para obter as caixas delimitadoras. Principalmente nesta linha. Apenas trackback o 't' variável e você vai ter o seu caixas delimitadoras.

Espero que ajude. Deixe-me saber se você precisar de mais esclarecimentos.

2021-11-20 08:35:20

A seguir thoose passos que eu poderia modificar o eval.py e obter a avaliação e caixas delimitadoras. Muito obrigado pela resposta, muito útil :D
Ramiro Sena

Em outros idiomas

Esta página está em outros idiomas

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