Yolov8在RK3588上进行自定义目标检测(一)

1.数据集和训练模型

项目地址:https://github.com/airockchip/ultralytics_yolov8.git

从github(htps:l/github.com/airockchip/ultralytics_yolov8)上获取yolov8模型。

下载项目:

复制代码
git clone https://github.com/airockchip/ultralytics_yolov8.git

安装:

Pip 使用 PyTorch>=1.7Python>=3.8 环境中安装包含所有要求的 ultralytics 包。

复制代码
pip install ultralytics

下载数据集:

数据集下载:从roboflow数据集网站下载火灾检测的数据集并导出为YOLO格式的标签文件(或者使用标注工具(如LabelImg或者LabelMe)进行数据标注,生成YOLO格式)。

导入数据集:

下载好的模型代码用Pycharm打开后,需要导入数据集进行训练。下面是数据集的结构:

修改配置文件:

dataset/data.yaml复制到ultralytics/cfg/datasets下,命名为my_dataset.yaml。

修改my_dataset.yaml的内容。

修改 ultralytics/cfg/model/v8/yolov8.yaml。

训练模型:

复制代码
yolo detect train data=C:/Users/WYX/Desktop/code/yolov8-main/ultralytics/cfg/datasets/my_dataset.yaml model=yolov8n.pt epochs=500 batch=32 imgsz=640 device=cpu

训练结果:

设置训练500次,在过去100个周期中没有观察到任何改进,训练提前停止。最佳结果出现在第217个周期上,最佳模型保存为best.pt,图像大小640。训练完成后会在yolov8-main/runs/detect/train weights目录下面产生一个best.pt的模型文件,即训练好的模型。

训练完成后分别进行预测和验证。

预测:

默认预测的数据图片在ultralytics/assets,需要预测前放入要预测的数据图片。

复制代码
yolo predict model=C:\Users\WYX\Desktop\code\yolov8-main\runs\detect\train\weights\best.pt

验证:

复制代码
yolo val model=C:\Users\WYX\Desktop\code\yolov8-main\runs\detect\train\weights\best.pt data=data=C:/Users/WYX/Desktop/code/yolov8-main/ultralytics/cfg/datasets/my_dataset.yaml batch=32

结果:

参考链接:

YOLOv8目标检测在RK3588部署全过程_yolov8 rk3588-CSDN博客

相关推荐
羞儿6 小时前
人体姿态估计系统知识框架
深度学习·计算机视觉·姿态估计·行为识别
大江东去浪淘尽千古风流人物10 小时前
【UniSim-SLAM】前馈SLAM的双流架构:Sim(3)统一位姿图、与经典SLAM及DPVO的三范式对比
深度学习·计算机视觉·3d·交互·人体姿态估计·eccv·3d重建
XiaoZhenHua9810 小时前
VisionPro多相机高速检测性能优化实战:从图像堆积到稳定运行
人工智能·计算机视觉·自动化
matlab代码10 小时前
基于matlab多尺度形态学提取眼前节组织【源码73期】
图像处理·人工智能·计算机视觉
richard_yuu11 小时前
AOI 实战第五篇:c_broken 漏检严重,V4 迭代背后的权衡
开发语言·深度学习·yolo
大江东去浪淘尽千古风流人物17 小时前
【LoMa】局部特征匹配重访:从LoMa-B到旋转不变LoMa-R的架构与工程实践
开发语言·深度学习·计算机视觉·r语言·视觉定位·sfm·局部特征匹配
XiaoZhenHua9820 小时前
C#工业机器视觉软件架构设计:从能运行的Demo到能够稳定跑产线的软件
计算机视觉·系统架构·自动化
YOLO数据集集合20 小时前
无人机航拍建筑与拆除废物检测数据集 | 建筑垃圾 无人机巡检 废物分类 资源回收 目标检测10042期
目标检测·分类·无人机
richard_yuu20 小时前
AOI 实战第六篇:.onnx 到 C++ 的最后一公里
c++·深度学习·yolo
这张生成的图像能检测吗20 小时前
(论文速读)基于两阶段多模式深度学习的轮胎表面缺陷自动检测及严重程度分类系统
人工智能·深度学习·计算机视觉·数据采集·检测系统·缺陷检测分类·轮胎缺陷