基于OpenVNO摆设YOLOv5-seg及时实例分割模子

分享
源码 2024-9-22 12:28:59 48 0 来自 中国
1. 简介

本文将先容在基于OpenVINO在英特尔独立显卡上摆设YOLOv5及时实例分割模子的全流程,并提供完备范例代码供读者使用。
YOLOv5是AI开辟者友好度最佳的框架之一,与别的YOLO系列相比:

  • 工程化水平好,工程应用时“坑”少
  • 文档详确友好,易读易懂
  • 既轻易在用户的数据集上重训练又轻易在差异的平台上举行摆设
  • 社区活跃度高(制止2023-1-21有34.7k GitHub星)
  • 项目演进速率快
  • 默认支持OpenVINO摆设
  • 在典范行业(制造业、农业、医疗、交通等)有广泛应用。
2022年11月22日,YOLOv5 v7.0版正式发布,成为YOLO系列中第一个支持及时实例分割(Real Time Instance Segmentation)的框架。今后,YOLOv5框架不但具有及时目的检测模子,还涵盖了图像分类和实例分割

2022年英特尔发布了代号为Alchemist的第一代消耗级锐炫™桌面独立显卡,OpenVINO™ 从2022.2版开始支持英特尔独立显卡,包罗英特尔® 数据中央 GPU Flex 系列和英特尔® 锐炫™系列。
2. 在英特尔独立显卡上摆设YOLOv5-seg模子的完备流程

在英特尔独立显卡上摆设YOLOv5-seg模子的完备流程紧张有三步:

  • 搭建YOLOv5开辟情况和OpenVINO摆设情况
  • 运行模子优化器(Model Optimizer)优化并转换模子
  • 调用OpenVINO Runtime API函数编写模子推理步伐,完成模子摆设
    本文将按照上述三个步调,依次详述
2.1 搭建YOLOv5开辟情况和OpenVINO摆设情况

迩来的YOLOv5 Github 代码仓,即YOLOv5 v7.0,已经将openvino-dev[onnx]写入requirement.txt文件,当执行pip install -r requirements.txt,会安装完YOLOv5开辟情况和OpenVINO摆设情况。
git clone https://github.com/ultralytics/yolov5  # clonecd yolov5
cd yolov5
pip install -r requirements.txt
2.2 验证YOLOv5开辟情况和OpenVINO摆设情况

执行完上述下令后,运行下令
python segment\predict.py --source data\images


执行结果如下图所示,阐明YOLOv5开辟情况和OpenVINO摆设情况已搭建乐成2.3 导出yolov5s-seg OpenVINO IR模子

使用下令:
python export.py --weights yolov5s-seg.pt --include onnx
得到yolov5s-seg ONNX格式模子:yolov5s-seg.onnx。然后运行下令:
mo -m yolov5s-seg.onnx --data_type FP16
得到yolov5s-seg IR格式模子:yolov5s-seg.xml和yolov5s-seg.bin
2.4 使用OpenVINO Runtime API编写yolov5s-seg推理步伐

整个推理步伐紧张有五个关键步调:
第一步:创建Core对象;
第二步:载入yolov5s-seg模子,并面向英特尔独立显卡编译模子
第三步:对图像数据举行预处理处罚
第四步:执行AI推理盘算
第五步:对推理结果举行后处理处罚,并可视化处理处罚结果。
整个代码框架如下所示:
您需要登录后才可以回帖 登录 | 立即注册

Powered by CangBaoKu v1.0 小黑屋藏宝库It社区( 冀ICP备14008649号 )

GMT+8, 2024-11-22 18:23, Processed in 0.186709 second(s), 32 queries.© 2003-2025 cbk Team.

快速回复 返回顶部 返回列表