Проблема с использованием openCV для загрузки сети из ONNX (python / pytorch)

Я пытаюсь загрузить обученную модель .onnx (из алгоритма передачи нейронного стиля) в cv2. Я видел, что есть

cv.dnn.readNetFromONNX() 

функция, но в cv2 такой функции нет. Кажется, я не могу импортировать или загружать opencv как cv, и поэтому не могу загрузить свою модель в cv2. Кто-нибудь знает решение?

Я в основном тренировал модель с https://github.com/pytorch/examples/blob/master/fast_neural_style/neural_style/neural_style.py#L122-L150 этот скрипт и сделал экспорт модели onnx, добавив

torch.onnx.export(style_model, dummy_input, "chipsoft_mod.onnx", verbose=True)

Теперь я хочу запустить обученную модель через читатель cv2, но я потерпел неудачу.

Также столкнулся с этой проблемой, еще одна веская причина использовать / изучать C++

A.Ametov 09.04.2019 10:45
Почему в Python есть оператор "pass"?
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
3
1
1 968
1

Ответы 1

Обновите свой opencv до более новой версии. Это должно помочь.

pip install opencv-python==4.1.0.25

Другие вопросы по теме