目录
安装openvino导出工具:
pip install openvino-dev会安装最新版,
我用的2021.04版本
pip install openvino-dev==2021.04
python调用代码:
model.eval()
dummy_input1 = torch.randn(1, 3, 64, 64).cuda()
onnx_file = "cls_skipnet.onnx"
torch.onnx.export(model, dummy_input1, onnx_file, opset_version=11, verbose=False, input_names=['images'], output_names=['output'])
export_openvino(model_cls.model, dummy_input1, onnx_file)
export_vino.py代码:
import os
import logging
import pkg_resources as pkg
import platform
from pathlib import Path
from subproces