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

相关推荐
大魔王(已黑化)34 分钟前
OpenCV —— 绘制图形
人工智能·opencv·计算机视觉
开开心心_Every1 小时前
多线程语音识别工具
javascript·人工智能·ocr·excel·语音识别·symfony
机器之心1 小时前
扣子开源全家桶,Apache 2.0加持,AI Agent又一次卷到起飞
人工智能
草堂春睡足1 小时前
【Datawhale AI夏令营】科大讯飞AI大赛(大模型技术)/夏令营:让AI理解列车排期表
人工智能·笔记
余俊晖1 小时前
GRPO强化学习缓解多模态大模型OCR任务的幻觉思路及数据生成思路
人工智能
sssammmm1 小时前
AI入门学习-模型评估示例讲解
人工智能·学习
小Tomkk1 小时前
AutoLabelImg:高效的数据自动化标注工具和下载
运维·人工智能·自动化
aneasystone本尊1 小时前
构建和使用 RAGFlow 的标签集
人工智能
柠檬味拥抱1 小时前
CI/CD与模型监控平台集成MLOps系统实现的全面路径
人工智能
人生都在赌2 小时前
MCP生态全景:2025年最值得关注的AI工具集成现状深度调研
人工智能·ai编程·mcp