Atlas200_YOLOv5_DVPP_AIPP
在Atlas200DK平台上基于DVPP和AIPP实现部署YOLOv5(v6.0)
· DVPP解码JPG,并Resize
· AIPP实现颜色转换:YUV420sp_U8 to RGB
效果:
前处理 | 推理 | 后处理 |
---|---|---|
5.21ms | 1.09ms | 4.41ms |
环境:
CANN版本:6.0.0.alpha003
OS版本:Ubuntu18.04
文件结构:
├─build
├─cvt_yolov5s
├─data
├─inc
├─model
├─res
└─src
使用:
-
下载yolov5,并切换版本:
git clone https://github.com/ultralytics/yolov5.git git checkout v6.0 pip install -r requirements.txt
-
模型转换pt --> onnx:
python export.py --include onnx --weigh