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

Question

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

cv.dnn.readNetFromONNX() 

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

Я в основном обучил модель с помощью https://github.com/pytorch/examples/blob/master/fast_neural_style/neural_style/neural_style.py этого сценария и сделал экспорт модели onnx, добавив

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

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

Answer 1

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

pip install opencv-python==4.1.0.25