Pytorch'u dağıtın .bir python betiğinde pth modeli

0

Soru

Yolact modelimi özel bir veri kümesi kullanarak başarılı bir şekilde eğittikten sonra, çıkarım sonuçlarından memnunum eval.py bu komutu anaconda terminalinden kullanma:

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

Şimdi bu çıkarımı anaconda terminalini kullanmak yerine kendi python betiğimden çalıştırmak istiyorum. Aşağıdaki kodla elde edilen web kamerası çerçevelerinde yapılan sınırlayıcı tespit kutularını elde edebilmek istiyorum. Herhangi bir fikir ?

import cv2

src = cv2.VideoCapture(0)

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

En eval.py kod burada Yolact deposunda https://github.com/dbolya/yolact/blob/master/eval.py

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

En iyi cevabı

0

Sahte kodu buraya senin için yazacağım.

Adım 1: Buradan başlayıp burada biten satırları kullanarak modeli yüklemeyi deneyin

Adım 2: Değerlendirme için bu işlevi kullanın. Cv2 yerine.imread, sadece çerçevenizi göndermeniz gerekiyor

Adım 3: Sınırlayıcı kutuları almak için bu işlevi izleyin. Özellikle bu çizgi. Sadece' t ' değişkenini takip edin ve sınırlayıcı kutularınızı alacaksınız.

Umarım yardımcı olur. Eğer daha fazla açıklama ihtiyacı varsa bana bildirin.

2021-11-20 08:35:20

Bu adımları takip ederek değiştirebilirim eval.py ve değerlendirme ve sınırlayıcı kutuları alın. Cevap için çok teşekkür ederim, çok yararlı: D
Ramiro Sena

Diğer dillerde

Bu sayfa diğer dillerde

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