yolov5训练数据,并运行在rv1126平台

拿到板子好多天了,终于完整跑通rk平台下的训练自已的数据用例了,以下是记录流程:

模型比较小,没有做预编译。从测试效果来看还行。

周末有空,再把效果优化一下,补上效果图。

1.(py38gpupt) rv1126@rv1126:~/ai/yolov5-7.0/yolov5-7.0$python train.py --data VOC.yaml --weights '' --cfg yolov5s.yaml --img 640 --batch-size 4 --epoch 10

2.可以用pc模拟测试:(py38gpupt) rv1126@rv1126:~/ai/yolov5-7.0/yolov5-7.0$ python detect.py --source 0 --weights ./runs/train/exp8/weights/best.pt --conf 0.2

3.修改yolo.py输出为3个出口 并导 (py38gpupt) rv1126@rv1126:~/ai/yolov5-7.0/yolov5-7.0$ gedit models/yolo.py &

4.导出(py38gpupt) rv1126@rv1126:~/ai/yolov5-7.0/yolov5-7.0$ python export.py --weights ./runs/train/exp8/weights/best.pt --img 640 --batch 1 --include onnx

5.切换环境到转换工具下:(py36rknn1.7.5) rv1126@rv1126:~/ai/rktoolkit$ python convert-onnx-to-rknn-copy.py

6.生成模型导入开发板,(py36rknn1.7.5) rv1126@rv1126:~/ai/rktoolkit$ adb push yolov5-7.0-onnx_tyd.rknn /demo/bin

7.修改应用程序,把模型改为 static char *model_path = "/demo/bin/yolov5-7.0-onnx_tyd.rknn";

8.应用程序使用:root@rv1126:/home/rv1126/ai/yolov5_object_recognize/build# adb push yolov5_object_recognize /demo/bin

相关推荐
阿里云云原生21 分钟前
山石网科×阿里云通义灵码,开启研发“AI智造”新时代
网络·人工智能·阿里云·ai程序员·ai程序员体验官
diemeng11191 小时前
AI前端开发技能变革时代:效率与创新的新范式
前端·人工智能
有Li1 小时前
跨中心模型自适应牙齿分割|文献速递-医学影像人工智能进展
人工智能
牧歌悠悠6 小时前
【深度学习】Unet的基础介绍
人工智能·深度学习·u-net
坚毅不拔的柠檬柠檬6 小时前
AI革命下的多元生态:DeepSeek、ChatGPT、XAI、文心一言与通义千问的行业渗透与场景重构
人工智能·chatgpt·文心一言
坚毅不拔的柠檬柠檬6 小时前
2025:人工智能重构人类文明的新纪元
人工智能·重构
jixunwulian6 小时前
DeepSeek赋能AI边缘计算网关,开启智能新时代!
人工智能·边缘计算
Archie_IT7 小时前
DeepSeek R1/V3满血版——在线体验与API调用
人工智能·深度学习·ai·自然语言处理
大数据追光猿7 小时前
Python应用算法之贪心算法理解和实践
大数据·开发语言·人工智能·python·深度学习·算法·贪心算法
灵感素材坊8 小时前
解锁音乐创作新技能:AI音乐网站的正确使用方式
人工智能·经验分享·音视频