ylov8的训练和预测使用(目标检测)

news/2024/7/10 2:56:09 标签: 目标检测, 人工智能, 计算机视觉

首先要配置文文件

1-配置数据集的yaml文件:

目录在ultralytics/cfg/datasets/下面:
例如我的:
(这里面的yaml文件在/ultralytics/cfg/datasets下面有很多,可以找几个参考一下)

path: /path/to/eye_datasets  # dataset root dir
train: train/images  # train images (relative to 'path') 4 images
val: valid/images  # val images (relative to 'path') 4 images
# test: images/test # test images (optional)
# Classes
names:
  0: eye

2- 配置.config/Ultralytics/settings.yaml 文件(/root/.config/Ultralytics/settings.yaml)

例如我的(更改了datasets_dir、weights_dir、runs_dir的路径):

settings_version: 0.0.4
datasets_dir: /share1/luli/yolov8/dataset/eye_datasets
weights_dir: /share1/luli/yolov8/eyeCodes/weights
runs_dir: /share1/luli/yolov8/eyeCodes/runs
uuid: 858bd79f1fda6637d7c2de0b0427e31d0157b9b3249c78658e02fe4956764daf
sync: true
api_key: ''
clearml: true
comet: true
dvc: true
hub: true
mlflow: true
neptune: true
raytune: true
tensorboard: true
wandb: true

训练代码

from ultralytics import YOLO
# 加载模型
model = YOLO('yolov8n.pt')  # 加载预训练模型(推荐用于训练)
# 使用1个GPU训练模型
results = model.train(data='/path/to/ultralytics/cfg/datasets/eyes.yaml', epochs=100, imgsz=640, device=[0])

训练完成:

在这里插入图片描述

注意事项:

import os, sys
sys.path.append("/share1/luli/yolov8")"/share1/luli/yolov8"这个里面下面的搜索范围添加到当前的文件里面。

from ..ultralytics import YOLO   ..表示在上上一级目录

http://www.niftyadmin.cn/n/5315025.html

相关文章

MT6785安卓核心板_联发科MTK6785/Helio G95/曦力G95核心板定制

MT6785安卓核心板是基于MT6785(Helio G95)处理器,具备八核处理器结构,包括2颗主频为2.05GHz的Cortex A76处理器和6颗主频为2.0GHz的Cortex A55处理器,以及六颗Cortex-A55处理器。而在GPU方面,采用了Arm Mali-G76 MC4,频…

统信UOS命令行设置未签名软件安装权限

原文链接:统信UOS命令行设置未签名软件安装权限 hello,大家好啊!今天我要给大家介绍的是在统信UOS操作系统上通过命令行设置安全中心应用安装权限的方法。在某些情况下,用户可能需要安装未经官方签名的软件包。虽然这可以提供更多…

设计不合理才是最大的问题

企业是否允许有多个管理员? 如果出现多个管理员,是否允许管理员登录? 某程序设计:如果企业有多个管理员,则管理员在登录时就会报错。 这么设计是否合理?肯定不同的人,各有各的看法。 从程序…

哈希表-散列表数据结构

1、什么是哈希表? 哈希表也叫散列表,哈希表是根据关键码值(key value)来直接访问的一种数据结构,也就是将关键码值(key value)通过一种映射关系映射到表中的一个位置来加快查找的速度,这种映射关系称之为哈希函数或者散列函数&…

使用Visual Leak Detector排查内存泄漏

目录 1、VLD工具概述 2、下载、安装VLD 2.1、下载VLD 2.2、安装VLD 3、VLD安装目录及文件说明

Postman工具使用一篇快速入门教程

文章目录 下载安装注册登录CollectionFolderRequestGet请求Post请求Header设置Response响应 EnvironmentsGlobal环境变量其他环境变量Collection变量变量使用同名变量的优先级 Postman内置变量Pre-request script和Test script脚本设置、删除和获取变量获取请求参数获取响应数据…

CNN和RNN的区别是什么?

CNN(卷积神经网络)和RNN(循环神经网络)是深度学习中的两种主要神经网络类型,它们在结构、功能和应用领域上有显著的区别: 核心结构与工作原理: CNN: CNN主要用于处理具有网格状拓扑结构的数据&#xff0c…

使用Ray Marching进行3D渲染

要使用Ray Marching渲染 3D 场景,我们需要为每个像素发射一条光线。通过选择屏幕上的一个点(一个像素),并从视点绘制一条射线,我们可以确定它是否击中一个对象,并决定应该在该像素上绘制什么。使用着色器&a…