(四)YOLOv5模型PTQ量化

YOLOv5模型介绍
Ultralytics YOLOv5 🚀 是由 Ultralytics 开发的尖端、最先进(SOTA)计算机视觉模型。YOLOv5 基于 PyTorch 框架构建,以其易用性、速度和准确性而闻名。它融合了广泛研究与开发中的见解和最佳实践,使其成为各种视觉 AI 任务的热门选择,包括目标检测、图像分割和图像分类。

(以上内容取自 Ultralytics YOLOv5 Github 仓库:https://github.com/ultralytics/yolov5)
模型下载与导出
1.Git下载YOLOv5;
2.安装依赖;
3.导出模型;
4.拷贝模型至“Desktop/dataset/”文件夹下;

模型验证与转换
1.启动 Open Explorer Docker环境
2.模型验证

3.图片转换
转换脚本如下:
运行测试如下:

4.模型转换
4.1编写yolov5n.yaml文件“Desktop/dataset/”文件夹下;
4.2进行模型转换;

