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博客

相关推荐
2501_924889555 小时前
商超高峰客流统计误差↓75%!陌讯多模态融合算法在智慧零售的实战解析
大数据·人工智能·算法·计算机视觉·零售
2501_924890529 小时前
商超场景徘徊识别误报率↓79%!陌讯多模态时序融合算法落地优化
java·大数据·人工智能·深度学习·算法·目标检测·计算机视觉
十八岁牛爷爷15 小时前
通过官方文档详解Ultralytics YOLO 开源工程-熟练使用 YOLO11实现分割、分类、旋转框检测和姿势估计(附测试代码)
人工智能·yolo·目标跟踪
阿杜杜不是阿木木16 小时前
什么?OpenCV调用cv2.putText()乱码?寻找支持中文的方法之旅
人工智能·opencv·计算机视觉
赴33516 小时前
图像边缘检测
人工智能·python·opencv·计算机视觉
机器视觉知识推荐、就业指导16 小时前
如何消除工业视觉检测中的反光问题
人工智能·计算机视觉·视觉检测
星期天要睡觉19 小时前
(纯新手教学)计算机视觉(opencv)实战八——四种边缘检测详解:Sobel、Scharr、Laplacian、Canny
人工智能·opencv·计算机视觉
2501_9247311120 小时前
智慧矿山误报率↓83%!陌讯多模态融合算法在矿用设备监控的落地优化
人工智能·算法·目标检测·视觉检测
nenchoumi311921 小时前
手持 Mid360 + RealSense 相机 + Jetson Orin 一体平台
人工智能·目标检测·计算机视觉·机器人·ros
lxmyzzs1 天前
【图像算法 - 21】慧眼识虫:基于深度学习与OpenCV的农田害虫智能识别系统
人工智能·深度学习·opencv·算法·yolo·目标检测·计算机视觉