超级会员免费看
本文介绍了如何将训练好的YOLOv5模型转换为OpenVINO格式，包括修改模型文件以适配OpenVINO、模型训练、ONNX转换以及在Windows上使用OpenVINO进行模型转换和CPU性能测试，最终实现高效的模型部署。
【玩转YOLOv5】YOLOv5的Openvino转换和部署
本文禁止转载！
1. YOLOv5环境配置：
可以看我之前写的几篇：
【小白CV教程】Pytorch训练YOLOv5并量化压缩（VOC格式数据集）
2. 修改模型文件：
由于YOLOv5的许多算子openvino仍然不支持，因此我们需要做出几点修改。首先我们需要将 Hardswish 激活函数替换掉，换成 LeackyReLU。
具体修改的地方有：
1. models/yolo.py
models/yolo.py：
# 第32行
self.act = nn.Hardswish(</
订阅专栏 解锁全文
扫一扫
1442
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
