使用YOLOv11实现水果类别检测：从数据到模型训练的全过程

发布于：2025-07-17 ⋅ 阅读:(12) ⋅ 点赞:(0)

项目背景

在日常生活中，水果的类别识别直接影响库存管理和质量控制。传统检测依赖人工，效率低下且主观。通过计算机视觉，我们可以自动化这个过程。YOLO系列模型以其速度和准确性闻名，这次我使用YOLOv11n进行训练，目标是检测31种水果类别，包括苹果、香蕉、橙子等。

数据集来源于公开来源，图像标注为YOLO格式，包含训练、验证和测试集。nc=31，names包括’fresh_apple’到’rotten_tomato’等。
在这里插入图片描述

方法步骤

数据准备：数据集已准备好，包含标注图像。

在这里插入图片描述

模型训练：使用训练脚本，配置YOLO数据。训练参数：epochs=50, batch=16, imgsz=640。

以下是训练脚本的主要代码：

from ultralytics import YOLO
import os
import glob
import torch

def main():
    model = YOLO('models/yolo11n.pt')
    yaml_path = 'data/data.yaml'
    results = model.train(data=yaml_path, epochs=50, batch=16, imgsz=640, name='yolo11n_fruit', project='fruit_results')
    print("Training completed. Results saved in:", results.save_dir)

if __name__ == '__main__':
    main()

训练命令：python train_fire_smoke_tassel.py

结果保存在fruit_results/yolo11n_fruit_20250716_210401。

在这里插入图片描述

模型验证：使用test_saved_model.py验证，加载best.pt模型。

以下是验证脚本的主要代码：

from ultralytics import YOLO
import os
import glob

def main():
    yaml_path = 'data/data.yaml'
    results_dir = 'fruit_results'
    latest_run = max(glob.glob(os.path.join(results_dir, 'yolo11n_fruit_*')), key=os.path.getmtime)
    model_path = os.path.join(latest_run, 'weights', 'best.pt')
    model = YOLO(model_path)
    metrics = model.val(data=yaml_path)
    print(f"Validation metrics: {metrics}")

if __name__ == '__main__':
    main()