HISI3559A YOLOV5训练部署全流程 转载

HISI3559A YOLOV5训练部署全流程
yolov5网络简介
https://zhuanlan.zhihu.com/p/172121380

hisi3559a开发板简介
CPU:

双核 ARM Cortex A73@1.8GHz，32KB I-Cache，64KB D-Cache /512KB L2 cache

双核 ARM Cortex A53@1.2GHz，32KB I-Cache，32KB D-Cache /256KB L2 cache

单核 ARM Cortex A53@1.2GHz，32KB I-Cache，32KB D-Cache /128KB L2 cache

支持 Neon 加速，集成 FPU 处理单元

GPU:

双核 ARM Mali G71@900MHz，256KB cache

支持 OpenCL 1.1/1.2/2.0

支持 OpenGL ES 3.0/3.1/3.2

智能视频分析：

提供视觉计算处理能力

四核 DSP@700MHz，32K I-Cache /32K IRAM/512KB DRAM

双核 NNIE@840MHz 神经网络加速引擎 INT8 4T算力

内置双目深度检测单元

yolov5网络模型训练
1）下载yolov5源码
a、git clone https://github.com/ultralytics/yolov5.git
b、git reset --hard 69be8e738

2）安装yolov5训练环境
a、conda create --name yolov5 python=3.7.9 -y
b、conda activate yolov5
c、修改requirements.txt，删除coremltools、onnx、scikit-learn前的”#“，增加一行“onnx-simplifier”
d、pip install -r requirements.txt

3）修改训练参数和模型结构
a、修改data/coco.yaml文件中类别数目、类别名、train/test/val的路径，按照自己的项目规划修改
b、修改models/yolov5s.yaml文件中类别数目
c、修改models/yolov5s.yaml中的网络结构，将focus层修改为卷积层，并设置stride为2

backbone:
# [from, number, module, args]
# [[-1, 1, Focus, [64, 3]], # 0-P1/2
[
[-1, 1, Conv, [64, 3, 2]],
[-1, 1, Conv, [128, 3, 2]], # 1-P2/4
[-1, 3, C3, [128]],
[-1, 1, Conv, [256, 3, 2]], # 3-P3/8
[-1, 9, C3, [256]],
[-1, 1, Conv, [512, 3, 2]], # 5-P4/16
[-1, 9, C3, [512]],
[-1, 1, Conv, [1024, 3, 2]], # 7-P5/32
[-1, 1, SPP, [1024, [5, 9, 13]]],
[-1, 3, C3, [1024, False]], # 9
]
或者用yolov5 release6.0版本已经去掉focus层，
backbone:
# [from, number, module, args]
[[-1, 1, Conv, [64, 6, 2, 2]], # 0-P1/2 <--- update
[-1, 1, Conv, [128, 3, 2]], # 1-P2/4
[-1, 3, C3, [128]],
[-1, 1, Conv, [256, 3, 2]], # 3-P3/8
[-1, 9, C3, [256]],
[-1, 1, Conv, [512, 3, 2]], # 5-P4/16
[-1, 9, C3, [512]],
[-1, 1, Conv, [1024, 3, 2]], # 7-P5/32
[-1, 1, SPP, [1024, [5, 9, 13]]],
[-1, 3, C3, [1024, False]], # 9
]

或者直接用
https://github.com/mahxn0/Yolov5-Hisi3559a-Train.git

4）启动模型训练
python train.py --data data/coco.yaml --cfg models/yolov5s.yaml --weights '' --batch-size 64 --img-size 416 --noautoanchor

5）模型导出
python models/export.py --weights weights/last.pt

6）模型简化
python -m onnxsim weights/last.onnx weights/simple.onnx (先安装onnx-simpler)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
torch 转caffe模型
确保已经搭建好caffe

cd yolov5_onnx2caffe

修改 convertCaffe.py 中路径

设置onnx_path（上面转换得到的简化后onnx模型），prototxt_path（caffe的prototxt保存路径），caffemodel_path（caffe的caffemodel保存路径）

python convertCaffe.py
1
2
3
4
5
6
7
得到转换后的caffemodel.

caffe模型转hisi3559 wk
本地linux转换环境搭建参考：https://blog.csdn.net/racesu/article/details/107045858

配置文件注意预处理顺序即可

后处理代码
https://github.com/mahxn0/Hisi3559A_Yolov5.git

多平台推理引擎,硬件图像处理工具 on the way…
————————————————
版权声明：本文为CSDN博主「朝风工作室」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/maxhn0/article/details/122417680