Windows下训练PyTorch版YOLOv5并用OpenVINO™ 部署 | 开发者实战

weixin_44119674

于 2023-04-24 10:12:01 发布

阅读量86

点赞数

分类专栏： deeep learning 文章标签： pytorch YOLO openvino

本文链接：https://blog.csdn.net/weixin_44119674/article/details/130336655

版权

deeep learning 专栏收录该内容

105 篇文章 6 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

本文详述如何在Windows环境下使用PyTorch训练YOLOv5模型，并通过OpenVINO™进行高效部署，实现快速的目标检测应用。

摘要由CSDN通过智能技术生成

OpenVINO™ 部署 YOLOV5

# ONNX转换为bin及xml文件
SUCCESS:☆☆☆！！！
python mo_onnx.py --input_model D:/DeepLearning/Project/yolov5-net-master/onnx_models/cursor320.onnx --output_dir D:/DeepLearning/Project/yolov5-net-master/openvino/FP16 --input_shape [1,3,320,320] --data_type FP16 --scale_values [127.5,127.5,127.5] --mean_values [127.5,127.5,127.5]