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

首先要配置文文件

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

目录在ultralytics/cfg/datasets/下面:

例如我的:

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

python 复制代码
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的路径):

python 复制代码
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

训练代码

python 复制代码
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])

训练完成:

注意事项:

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

from ..ultralytics import YOLO   ..表示在上上一级目录
相关推荐
yusaisai大鱼1 小时前
TensorFlow如何调用GPU?
人工智能·tensorflow
LNTON羚通1 小时前
摄像机视频分析软件下载LiteAIServer视频智能分析平台玩手机打电话检测算法技术的实现
算法·目标检测·音视频·监控·视频监控
珠海新立电子科技有限公司3 小时前
FPC柔性线路板与智能生活的融合
人工智能·生活·制造
IT古董3 小时前
【机器学习】机器学习中用到的高等数学知识-8. 图论 (Graph Theory)
人工智能·机器学习·图论
曼城周杰伦4 小时前
自然语言处理:第六十三章 阿里Qwen2 & 2.5系列
人工智能·阿里云·语言模型·自然语言处理·chatgpt·nlp·gpt-3
余炜yw4 小时前
【LSTM实战】跨越千年,赋诗成文:用LSTM重现唐诗的韵律与情感
人工智能·rnn·深度学习
莫叫石榴姐5 小时前
数据科学与SQL:组距分组分析 | 区间分布问题
大数据·人工智能·sql·深度学习·算法·机器学习·数据挖掘
如若1235 小时前
利用 `OpenCV` 和 `Matplotlib` 库进行图像读取、颜色空间转换、掩膜创建、颜色替换
人工智能·opencv·matplotlib
YRr YRr5 小时前
深度学习:神经网络中的损失函数的使用
人工智能·深度学习·神经网络
ChaseDreamRunner5 小时前
迁移学习理论与应用
人工智能·机器学习·迁移学习