yolo11n-cls
时间: 2025-07-01 14:05:27 浏览: 6
### YOLOv8n-cls 模型介绍
YOLOv8n-cls 是 YOLOv8 系列中的一种轻量级图像分类模型,适用于资源受限的场景。该模型在 ImageNet 数据集上进行了预训练,能够快速准确地对图像进行分类[^1]。由于其较小的模型规模,YOLOv8n-cls 特别适合部署在边缘设备或嵌入式系统中。
与其他 YOLOv8 分类模型(如 YOLOv8s-cls、YOLOv8m-cls 等)相比,YOLOv8n-cls 具有更快的推理速度和更低的计算需求,但精度略有下降。这使得它成为实时应用和低功耗设备的理想选择。
### 使用教程
#### 1. 加载预训练模型
可以使用以下代码加载预训练的 YOLOv8n-cls 模型:
```python
from ultralytics import YOLO
# 加载预训练模型
model = YOLO("yolov8n-cls.pt")
```
#### 2. 进行单张图像分类
加载模型后,可以直接对图像进行分类预测:
```python
# 对图像进行预测
results = model("path/to/image.jpg")
# 输出预测结果
for result in results:
print(result.probs) # 打印分类概率
```
#### 3. 自定义数据集训练
如果需要在自定义数据集上进行微调,可以使用如下代码:
```python
# 加载模型并准备训练
model = YOLO("yolov8n-cls.pt") # 推荐用于训练
# 开始训练
results = model.train(
data="path/to/dataset", # 数据集路径
epochs=100, # 训练轮数
imgsz=64, # 输入图像大小
device="0", # 使用的 GPU 设备
batch=32, # 批处理大小
project="runs/train", # 训练结果保存路径
name="exp" # 实验名称
)
```
#### 4. 多 GPU 训练支持
YOLOv8n-cls 支持多 GPU 训练以加速收敛过程:
```python
# 在多个 GPU 上训练模型
results = model.train(
data="path/to/dataset",
epochs=100,
imgsz=64,
device="0, 1", # 使用两个 GPU
batch=64,
project="runs/train",
name="multi_gpu_exp"
)
```
#### 5. 调整优化器与训练策略
可以通过设置 `optimizer` 和 `patience` 参数来调整优化器类型和早停策略:
```python
# 设置优化器为 Adam,并启用早停
results = model.train(
data="path/to/dataset",
epochs=300,
imgsz=64,
device="0",
batch=32,
optimizer="Adam",
patience=50,
project="runs/train",
name="adam_optimized"
)
```
###
阅读全文
相关推荐

















