例子
介绍
使用python实现基于onnxruntime推理框架的深度学习模型的推理功能。 可以将onnx模型转换为大多数主流的深度学习推理框架模型,因此您可以在部署模型之前测试onnx模型是否正确。
注意:此处的模型由pytorch 1.6训练,并由onnx 1.8.1转换
要求
onnx == 1.8.1
onnxruntime == 1.7.0或onnxruntime-gpu == 1.3.0
opencv-python == 4.2.0.32
运行演示
该演示以main_xxx_.py格式命名。您可以使用以下示例运行代码。
python main_pose_.py --det_model_path weights/yolov5s.onnx \
--pose_model_path data/det/zidane.jpg \
-
1