当前位置:   article > 正文

模型推理:YOLOv5 目标检测_yolov5 寒武纪

yolov5 寒武纪

一、内容和目标

1. 实验内容
  1. 本实验主要介绍基于寒武纪 MLU370 MagicMind 平台的YOLOv5m (PyTorch, Python, INT8)目标检测推理应用的开发方法,其中性能优化操作包括:INT8量化、集成后处理自定义算子。

  2. 基于 YOLOv5m 检测网络和寒武纪 MLU370 MagicMind 平台,您可以读取本地图像数据作为输入,对图像数据中的目标物体进行时识别,得到物体的在图像中的坐标、置信度以及类别。

2. 实验目标
  1. 掌握使用寒武纪推理引擎 MagicMind 进行模型量化推理及融合算子的基本方法。

  2. 理解 YOLOv5m 模型的整体网络结构及其开发调试细节。

二、前置知识

1. 寒武纪软硬件平台介绍

  • 硬件:寒武纪 MLU370 AI 加速卡

  • 框架:PyTorch 1.6、MagicMind 0.14.0

2. MagicMind 介绍

MagicMind 是⾯向寒武纪 MLU370 (寒武纪处理器,简称MLU) 的推理加速引擎。 MagicMind 能将人工智能框架(TensorFlow,PyTorch 等)训练好的算法模型转换成 MagicMind 统⼀计算图表⽰,并提供端到端的模型优化、代码⽣成以及推理业务部署能⼒。用户无需过多关注底层硬件细节,只需专注于推理业务开发。

获取更多有关 Magic

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/秋刀鱼在做梦/article/detail/761717
推荐阅读
相关标签
  

闽ICP备14008679号