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