Развертывание модели pytorch .pth в скрипте python

0

Вопрос

После успешного обучения моей модели yolact с использованием пользовательского набора данных я доволен результатами вывода, полученными eval.py используя эту команду из терминала anaconda:

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

Теперь я хочу запустить этот вывод из моего собственного скрипта python вместо использования терминала anaconda. Я хочу иметь возможность получать ограничительные рамки обнаружений, сделанных на кадрах веб-камеры, полученных с помощью этого кода ниже. Есть идеи ?

import cv2

src = cv2.VideoCapture(0)

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

В eval.py код находится здесь, в репозитории Yolact https://github.com/dbolya/yolact/blob/master/eval.py

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

Лучший ответ

0

Я просто напишу здесь псевдокод для вас.

Шаг 1: Попробуйте загрузить модель, используя строки, начинающиеся здесь и заканчивающиеся здесь

Шаг 2: Используйте эту функцию для оценки. Вместо cv2.imread вам просто нужно отправить свой кадр

Шаг 3: Следуйте этой функции, чтобы получить ограничительные рамки. Особенно эта строчка. Просто отследите переменную 't' , и вы получите свои ограничительные рамки.

Надеюсь, это поможет. Дайте мне знать, если вам понадобятся дополнительные разъяснения.

2021-11-20 08:35:20

Следуя этим шагам, я мог бы изменить eval.py и получите оценку и ограничительные рамки. Большое вам спасибо за ответ, очень полезный :D
Ramiro Sena

На других языках

Эта страница на других языках

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