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

相关推荐
云烟成雨TD几秒前
Spring AI 1.x 系列【26】结构化输出执行流程
java·人工智能·spring
清空mega2 分钟前
动手学深度学习——物体检测
人工智能
蓝色的杯子2 分钟前
OpenClaw一文详细了解-手搓OpenClaw-4 Tool Runtime
人工智能·python
波动几何3 分钟前
你好,我是 Adaptive Skill Stack
人工智能
皮皮学姐分享-ppx5 分钟前
1447上市公司数字化转型速度的计算(2000-2022年)
大数据·人工智能
张二娃同学12 分钟前
Claude Code 使用教程:下载安装、CC Switch 配置、MiniMax API 获取与启动实操
人工智能·windows·深度学习·github·claude code
yitian_hm13 分钟前
RAG实战:从原理到代码,构建企业级知识库问答系统
人工智能
AI品信智慧数智人14 分钟前
文旅景区小程序集成数字人智能语音交互系统,山东品信解锁AI伴游新玩法✨
人工智能·小程序
Rick199315 分钟前
LangChain和spring ai是什么关系?
人工智能·spring·langchain
AI创界者17 分钟前
【首发】LTX-2.3-VBVR 增强版发布:8G 显存解锁无限时长,视频一致性与运动精度跨越式升级!
人工智能