onnx 测试_亲测有效 | OpenVINO支持ONNX格式文件直接读取了

原标题：亲测有效 | OpenVINO支持ONNX格式文件直接读取了

微信公众号：OpenCV学堂

关注获取更多计算机视觉与深度学习知识

01

函数支持

OpenVINO 2020R04版本支持ONNX格式文件的直接读取了，使用的函数还是跟之前读取IR文件函数保持一致，只是第二个参数缺省为空时候，就会尝试读取ONNX格式文件。相关的函数与参数解释如下：

CNNNetwork InferenceEngine::Core::ReadNetwork(

conststd:: string& modelPath,

conststd:: string& binPath = {}

) const

其中

modelPath 表示模型输入路径，( .xml 或者 .onnx)

binPath 表示IR格式化数据路径(*.bin)，如果为空，尝试读取跟modelPath同名的bin文件，如果失败就会尝试直接加载IR文件。

02

ResNet模型转换

这里我使用pytorch中torchvision自带的ResNet18预训练模型，首先把它从pth转换为ONNX格式，转换的脚本如下：

model = torchvision.models.resnet18(pretrained=True).eval

dummy_input = torch.randn((1, 3, 224, 224))

torch.onnx. export(model, dummy_input, "resnet18.onnx")

转换为IR中间文件格式，参见下面的