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

相关推荐
IT_陈寒3 分钟前
闭包陷阱让我加了两天班,JavaScript你真行
前端·人工智能·后端
wumingxiaoyao5 分钟前
从 0 开始学 AI:第 2 课,AI、机器学习、深度学习和大模型是什么关系?
人工智能·深度学习·机器学习·ai·大模型·llm
2601_9626838919 分钟前
治理遗留系统中的“生肉 SQL”:一次用多模型协作优化慢查询的实战复盘
数据库·人工智能·sql
踩着两条虫20 分钟前
可视化 vs 终端 vs 云端:VTJ.PRO、Claude Code、Codex 三强横评
前端·vue.js·人工智能·低代码·架构
字节跳动视频云技术团队22 分钟前
Agentic 范式下的视频画质优化:火山引擎的新路径
人工智能·音视频开发
欢子1 小时前
Android 端接入AI
人工智能
酱学编程1 小时前
【从零到一实现一个 AI Agent 框架 · 第四篇】04. 任务规划:拆解复杂目标 -
服务器·网络·数据库·人工智能
手写码匠1 小时前
手写 AI 上下文压缩系统:从零实现 Prompt 压缩与选择性上下文管理
人工智能·深度学习·算法·aigc
机器之心1 小时前
世界模型评测的最大盲区,被这个新基准捅破了
人工智能·openai
机器之心1 小时前
Claude、GLM、GPT谁才是真正的AI软件工程师?首个持续更新Visual Spec-to-App Benchmark发布
人工智能·openai