Я пытаюсь загрузить обученную модель .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, но я потерпел неудачу.






Обновите свой opencv до более новой версии. Это должно помочь.
pip install opencv-python==4.1.0.25
Также столкнулся с этой проблемой, еще одна веская причина использовать / изучать C++