U版YOLO V8项目使用-Win11系统

U版YOLO V8项目使用-Win11系统

创建激活虚拟环境

  1. 为Yolov8创建所需要的虚拟环境

    conda create --name Yolov8 python=3.11.9

  1. 激活YOLO v8虚拟环境

    activate Yolov8

  2. 通过桌面程序查看创建的结果

拉取项目源码

https://github.com/ultralytics/ultralytics?tab=readme-ov-file

拉取项目到本地并使用Pycharm打开项目。

git clone https://github.com/ultralytics/ultralytics.git

安装依赖

  • 根据官网的提示使用可编辑模式安装所需要的库

    Clone the ultralytics repository

    git clone https://github.com/ultralytics/ultralytics

    Navigate to the cloned directory

    cd ultralytics

    Install the package in editable mode for development

    pip install -e .

预测与训练

输入命令执行测试文件使用命令行的方式执行单张图片的检测命令。

yolo predict model=yolov8s.pt source=D:/Git-res/DeepLearing/DL_01/YOLOV8/ultralytics-8.2.60/ultralytics/assets/bus.jpg

(Yolov8) D:\Git-res\DeepLearing\DL_01\YOLOV8\ultralytics-8.2.60

yolo predict model=yolov8s.pt source=D:/Git-res/DeepLearing/DL_01/YOLOV8/ultralytics-8.2.60/ultralytics/assets/bus.jpg

Downloading https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8s.pt to 'yolov8s.pt'...

100%|███████████████████████████████████████████████████████████████████████████████████████████████████| 21.5M/21.5M [00:13<00:00, 1.70MB/s]

Ultralytics YOLOv8.2.60 🚀 Python-3.11.9 torch-2.3.1+cu118 CUDA:0 (NVIDIA GeForce RTX 4060 Laptop GPU, 8188MiB)

YOLOv8s summary (fused): 168 layers, 11,156,544 parameters, 0 gradients, 28.6 GFLOPs

image 1/1 D:\Git-res\DeepLearing\DL_01\YOLOV8\ultralytics-8.2.60\ultralytics\assets\bus.jpg: 640x480 4 persons, 1 bus, 103.3ms

Speed: 9.5ms preprocess, 103.3ms inference, 60.7ms postprocess per image at shape (1, 3, 640, 480)

Results saved to runs\detect\predict

💡 Learn more at https://docs.ultralytics.com/modes/predict

使用coco128数据集和官网提供的命令模式进行训练。

官方提供的三种常用的训练命令。

# Build a new model from YAML and start training from scratch
yolo detect train data=coco8.yaml model=yolov8n.yaml epochs=100 imgsz=640

# Start training from a pretrained *.pt model
yolo detect train data=coco8.yaml model=yolov8n.pt epochs=100 imgsz=640

# Build a new model from YAML, transfer pretrained weights to it and start training
yolo detect train data=coco8.yaml model=yolov8n.yaml pretrained=yolov8n.pt epochs=100 imgsz=640

普通的训练模式

yolo detect train data=D:/ultralytics/ultralytics/cfg/datasets/VOC-new.yaml model=yolov8s.pt epochs=50 imgsz=640 batch=8 workers=4

断点续存模式

yolo detect train data=D:/ultralytics/ultralytics/cfg/datasets/VOC-new.yaml model=D:/ultralytics/runs/detect/train/weights/best.pt epochs=300 imgsz=640 batch=16 workers=4 resume

根据上面提供的参考格式:我们使用项目中默认提供好了的coco128.yaml文件进行训练。(方便起见训练轮数设置为50batch设置为8进行训练)

yolo detect train data=coco128.yaml  model=yolov8n.pt epochs=50 imgsz=640 batch=8 workers=4

新版的项目会自己下载数据集以及需要的预训练模型。之后就开始进行训练。

我自己在训练的时候发现,在训练的过程中当训练到41轮的时候会关闭数据增强训练。

评估训练出的网络模型

在官方文档中给出了两种模型评估方式。

  1. yolo detect val model=yolov8n.pt # val official model yolo detect val model=path/to/best.pt # val custom model

  2. 带有参数的模型评估方式

    yolo val model=yolov8n.pt data=coco8.yaml imgsz=640 batch=16 conf=0.25 iou=0.6 device=0

参考格式:

yolo detect val model=D:\Git-res\DeepLearing\DL_01\YOLOV8\ultralytics-8.2.60\runs\detect\train\weights\best.pt
data=D:/ultralytics/ultralytics/cfg/datasets/VOC-new.yaml

个人实验:我们在coco8数据集上进行实验验证集的过程。使用的命令行命令如下所示。

yolo detect val model=D:\Git-res\DeepLearing\DL_01\YOLOV8\ultralytics-8.2.60\runs\detect\train\weights\best.pt data=coco8.yaml

总结以上就是使用命令行cli方式跑YOLOV8代码的执行流程。

相关推荐
goomind1 小时前
YOLOv8实战木材缺陷识别
人工智能·yolo·目标检测·缺陷检测·pyqt5·木材缺陷识别
吾门2 小时前
YOLO入门教程(三)——训练自己YOLO11实例分割模型并预测【含教程源码+一键分类数据集 + 故障排查】
yolo·分类·数据挖掘
向阳逐梦5 小时前
ROS机器视觉入门:从基础到人脸识别与目标检测
人工智能·目标检测·计算机视觉
Suyuoa8 小时前
附录2-pytorch yolov5目标检测
python·深度学习·yolo
思通数科多模态大模型1 天前
10大核心应用场景,解锁AI检测系统的智能安全之道
人工智能·深度学习·安全·目标检测·计算机视觉·自然语言处理·数据挖掘
红色的山茶花1 天前
YOLOv8-ultralytics-8.2.103部分代码阅读笔记-block.py
笔记·深度学习·yolo
sp_fyf_20241 天前
【大语言模型】ACL2024论文-18 MINPROMPT:基于图的最小提示数据增强用于少样本问答
人工智能·深度学习·神经网络·目标检测·机器学习·语言模型·自然语言处理
思通数科AI全行业智能NLP系统1 天前
六大核心应用场景,解锁AI检测系统的智能安全之道
图像处理·人工智能·深度学习·安全·目标检测·计算机视觉·知识图谱
非自律懒癌患者1 天前
Transformer中的Self-Attention机制如何自然地适应于目标检测任务
人工智能·算法·目标检测
unix2linux1 天前
YOLO v5 Series - Image & Video Storage ( Openresty + Lua)
yolo·lua·openresty