简介这份资源面向计算机、人工智能、自动化等专业的在校学生与教师提供一套基于YOLOv8的果园成熟果实自动计数完整方案可用于毕业设计、课程设计或大作业。压缩包共8个文件约15.91MB包含3个Python脚本、3个模型权重文件与2个说明文本分别对应可视化界面、模型训练与视频检测等核心环节并附有完整数据集与部署教程简单配置即可运行。项目可输出核心指标曲线图、混淆矩阵、F1分数曲线、精确率-召回率曲线、验证集预测结果及标签分布图便于答辩展示与结果分析。目前已有30人学习关注。读者可直接获得经过测试的源码、预训练权重与可视化页面既能快速复现果实计数效果也能在此基础上修改扩展功能适合作为毕设保底方案或入门目标检测的实践素材。1. 果园果实自动计数这套 YOLOv8 毕设资源到底能不能直接跑果园里数果子这件事人工干一天下来眼睛发花数错率还高。这套资源的核心就是用 YOLOv8 做成熟果实的目标检测再把检测框转成计数结果配了一个可视化界面数据集和部署教程都打包在里面。拿到手之后不需要自己从零标注、从零搭界面解压完按教程走一遍就能跑起来。适合两类人一类是毕设或课程设计时间紧、需要完整可演示系统的学生另一类是想快速验证 YOLOv8 在农业计数场景下效果的从业者。资源里包含源码、可视化界面、完整数据集和部署教程技术栈是 Python YOLOv8 图形界面框架整体走的是“检测即计数”的路线不涉及复杂的多目标跟踪。下面按实际拆包和复现的顺序把环境配置、数据组织、训练调参、界面联调和常见翻车点逐个讲清楚。2. 环境配置与依赖安装从零到能跑通第一张推理图2.1 为什么优先选 CPU 版先跑通再换 GPU很多人拿到 YOLOv8 项目第一反应是装 CUDA、配 cuDNN结果卡在驱动版本上耗掉一整天。我的建议是先用 CPU 版本把推理和界面跑通确认代码逻辑没问题再切 GPU 训练。原因很直接CPU 版安装简单不会因为显卡驱动和 PyTorch 版本不匹配导致 import 就报错。等推理通了再装 GPU 版 PyTorch 做训练这时候即使出问题你也能确定是环境问题而不是代码问题。常见做法是建一个独立虚拟环境避免和系统 Python 混在一起。Ubuntu 20.04 和 Windows 10/11 都适用下面以 conda 为例。# 创建虚拟环境Python 版本建议 3.9 或 3.10 conda create -n yolov8_fruit python3.10 -y conda activate yolov8_fruit # 安装 CPU 版 PyTorch先跑通推理用 pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu # 安装 ultralytics 和界面依赖 pip install ultralytics opencv-python pillow pyqt5这里torch和torchvision走的是 CPU 源装完大概几百 MB。ultralytics是 YOLOv8 的官方库版本建议锁在 8.0.x 到 8.1.x 之间太新的版本有时会改 API 导致项目里的推理脚本报参数错误。pyqt5是可视化界面的依赖如果项目用的是 tkinter 就换成tkinterPython 自带不用额外装。装完之后跑一句yolo predict modelyolov8n.pt sourcehttps://ultralytics.com/images/bus.jpg验证环境能输出检测结果图就说明基础环境没问题。2.2 GPU 训练环境的切换与版本对齐CPU 推理通了之后如果要训练自己的数据集GPU 是绕不开的。这时候把 CPU 版 PyTorch 卸掉换成对应 CUDA 版本的 GPU 版。关键点是先确认显卡驱动支持的 CUDA 最高版本再选 PyTorch 对应的 CUDA 版本不要反过来。# 查看显卡驱动和 CUDA 支持版本 nvidia-smi # 卸载 CPU 版安装 GPU 版以 CUDA 11.8 为例 pip uninstall torch torchvision -y pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118 # 验证 GPU 是否可用 python -c import torch; print(torch.cuda.is_available())nvidia-smi右上角显示的 CUDA Version 是驱动支持的最高版本不是当前安装的版本。比如显示 12.2你可以装 CUDA 11.8 的 PyTorch也可以装 12.1 的但不能装 12.3 的。torch.cuda.is_available()返回 True 才算成功。如果返回 False先检查驱动版本再检查 PyTorch 和 CUDA 是否匹配最后看是不是装成了 CPU 版。这一步没有后悔药版本对不上就是跑不了 GPU。注意如果项目里 requirements.txt 锁定了 torch 版本优先按它来不要自己升级。版本冲突是 YOLOv8 项目里最常见的翻车原因之一。3. 数据集组织与标注格式转换让 YOLOv8 认得你的果园图片3.1 目录结构与 data.yaml 的对应关系YOLOv8 对数据集的目录结构有固定要求不是随便放几个文件夹就能训练。标准结构是 images 和 labels 分开各自下面再分 train、val、test。资源里如果已经带了完整数据集先检查目录是不是这个结构不是的话按下面调整。fruit_dataset/ ├── images/ │ ├── train/ # 训练集图片 │ ├── val/ # 验证集图片 │ └── test/ # 测试集图片可选 ├── labels/ │ ├── train/ # 训练集标注 txt │ ├── val/ # 验证集标注 txt │ └── test/ # 测试集标注 txt └── data.yaml # 数据集配置文件data.yaml是训练时的入口配置内容必须和实际路径对应。常见写法如下path: ./fruit_dataset # 数据集根目录 train: images/train # 训练集相对路径 val: images/val # 验证集相对路径 test: images/test # 测试集相对路径没有就删掉这行 nc: 1 # 类别数果园成熟果实通常只检测一类 names: [ripe_fruit] # 类别名称顺序要和标注里的 class_id 对应nc是类别数量如果只检测成熟果实就是 1。names列表的顺序很重要标注文件里的 class_id 是 0 就对应列表第一个名字。如果数据集里还有未成熟果实或其他类别nc和names都要相应增加。路径可以用相对路径也可以用绝对路径相对路径是相对于你执行训练命令时所在的目录不是相对于 data.yaml 文件本身这一点容易搞混。3.2 标注格式转换Labelme 转 YOLO 的脚本与边界处理如果数据集里的标注是 Labelme 生成的 JSON 格式需要转成 YOLO 的 txt 格式。转换的核心是把多边形或矩形坐标归一化到 0 到 1 之间格式是class_id x_center y_center width height。下面是一个可直接用的转换脚本。import json import os from pathlib import Path def labelme_to_yolo(json_dir, output_dir, class_map): json_dir: Labelme JSON 文件目录 output_dir: 输出 YOLO txt 的目录 class_map: 类别名到 id 的映射如 {ripe_fruit: 0} json_dir Path(json_dir) output_dir Path(output_dir) output_dir.mkdir(parentsTrue, exist_okTrue) for json_file in json_dir.glob(*.json): with open(json_file, r, encodingutf-8) as f: data json.load(f) img_w data[imageWidth] img_h data[imageHeight] lines [] for shape in data[shapes]: label shape[label] if label not in class_map: continue # 跳过未定义类别避免训练时报错 class_id class_map[label] points shape[points] # 取多边形或矩形的外接框 xs [p[0] for p in points] ys [p[1] for p in points] x_min, x_max min(xs), max(xs) y_min, y_max min(ys), max(ys) # 归一化并计算中心点和宽高 x_center (x_min x_max) / 2.0 / img_w y_center (y_min y_max) / 2.0 / img_h width (x_max - x_min) / img_w height (y_max - y_min) / img_h # 边界裁剪防止坐标超出 0-1 范围 x_center min(max(x_center, 0.0), 1.0) y_center min(max(y_center, 0.0), 1.0) width min(max(width, 0.0), 1.0) height min(max(height, 0.0), 1.0) lines.append(f{class_id} {x_center:.6f} {y_center:.6f} {width:.6f} {height:.6f}) # 写入同名 txt txt_path output_dir / (json_file.stem .txt) with open(txt_path, w, encodingutf-8) as f: f.write(\n.join(lines)) if __name__ __main__: labelme_to_yolo( json_dir./raw_annotations, output_dir./fruit_dataset/labels/train, class_map{ripe_fruit: 0} )这个脚本做了三件事读取 Labelme JSON、把多边形转成外接矩形并归一化、写入 YOLO 格式 txt。class_map参数控制类别映射数据集里有什么类别就加什么没在 map 里的标注会被跳过避免训练时出现未知 class_id 导致报错。归一化后的坐标做了 0 到 1 的裁剪防止标注时点超出图片边界导致训练异常。转换完记得抽查几个 txt 文件确认坐标数值在合理范围内不要出现大于 1 或负数的情况。注意图片和标注文件名必须一一对应除了扩展名不同。如果图片是img_001.jpg标注必须是img_001.txt放在 labels 对应目录下。名字对不上训练时会被当成无标注图片跳过。4. 模型训练与参数调优从默认配置到果园场景适配4.1 训练命令与关键参数含义YOLOv8 的训练入口很简洁一条命令就能启动。但默认参数不一定适合果园果实这种密集小目标场景需要根据实际情况调整。# 基础训练命令 yolo detect train \ data./fruit_dataset/data.yaml \ modelyolov8n.pt \ epochs100 \ imgsz640 \ batch16 \ lr00.01 \ patience20 \ project./runs/train \ namefruit_expmodel参数决定用哪个预训练权重。yolov8n.pt是最小的 nano 版本速度快但精度低如果果实目标小且密集建议换成yolov8s.pt或yolov8m.pt。epochs是训练轮数100 轮起步数据集小的话可以加到 200。imgsz是输入图片尺寸640 是默认值果实占画面比例小的话可以提到 1280但显存占用会明显增加。batch是批大小显存不够就往下调从 16 降到 8 再到 4。lr0是初始学习率默认 0.01数据集小可以降到 0.001 防止过拟合。patience是早停轮数20 轮内验证指标没提升就自动停省时间。训练过程中会在runs/train/fruit_exp下生成权重文件、损失曲线和验证结果。重点关注results.csv里的metrics/mAP50和metrics/mAP50-95前者是 IoU 0.5 时的平均精度后者是 0.5 到 0.95 的平均值。果园计数场景下 mAP50 到 0.85 以上基本可用低于 0.7 就要检查数据集标注质量或增加数据量。4.2 小目标密集场景的参数调整策略果园果实往往在画面里占比较小而且成熟果实可能成簇出现默认的 640 输入尺寸和 anchor 配置不一定最优。几个实际有效的调整方向参数默认值果园场景建议原因imgsz640960 或 1280提高小目标分辨率modelyolov8nyolov8s 或 yolov8m更大模型对小目标更敏感batch168 或 4大输入尺寸下显存不够lr00.010.005 或 0.001小数据集防过拟合mosaic1.00.5 或关闭果实密集时 mosaic 可能引入干扰close_mosaic1020 或 30最后阶段关闭增强稳定收敛mosaic是 YOLOv8 默认开启的数据增强把四张图拼成一张训练。果实计数场景下如果图片本身果实就很密集mosaic 可能让模型学到错误的上下文关系适当降低比例或最后几十轮关掉。close_mosaic控制最后多少轮关闭 mosaic设大一点让模型在真实分布上多收敛几轮。训练完之后用验证集跑一遍评估命令是yolo detect val model./runs/train/fruit_exp/weights/best.pt data./fruit_dataset/data.yaml。看输出的混淆矩阵和各类别 AP如果只有一类重点看漏检和误检的比例。漏检多说明模型太保守可以降低置信度阈值误检多说明模型把背景当成了果实需要增加负样本或提高置信度阈值。5. 可视化界面联调与计数逻辑让检测框变成数字5.1 界面调用推理的代码结构与参数传递可视化界面通常是一个独立的 Python 脚本通过按钮触发图片选择、推理和结果显示。核心逻辑是加载训练好的权重对选中图片做推理把检测框画在图上同时统计框的数量作为计数结果。下面是一个典型的界面推理函数。from ultralytics import YOLO import cv2 class FruitCounter: def __init__(self, model_path, conf_threshold0.5): self.model YOLO(model_path) self.conf_threshold conf_threshold def detect_and_count(self, image_path): # 执行推理conf 控制置信度阈值 results self.model(image_path, confself.conf_threshold) # 获取检测框数量 boxes results[0].boxes count len(boxes) if boxes is not None else 0 # 绘制结果图 annotated results[0].plot() return annotated, count # 界面里调用 counter FruitCounter(./runs/train/fruit_exp/weights/best.pt, conf_threshold0.5) img, num counter.detect_and_count(./test_images/orchard_01.jpg) cv2.imwrite(./output/result_01.jpg, img) print(f检测到成熟果实数量{num})conf_threshold是置信度阈值默认 0.5。这个参数直接决定计数结果调高会减少误检但可能漏检调低会检出更多但可能把背景算进去。果园场景下建议先在验证集上试几个值看哪个阈值下计数误差最小。results[0].boxes是检测框对象len()就是数量。results[0].plot()返回画好框的图片可以直接显示在界面上或保存。界面部分如果是 PyQt5把detect_and_count绑定到按钮的点击事件上图片显示用 QLabel 的 setPixmap。如果是 tkinter用 Canvas 或 Label 显示。关键是把推理放在子线程里不要在主线程跑否则界面会卡死。这一点在演示的时候特别明显老师或评委看到界面无响应会直接扣分。5.2 计数逻辑的边界处理与结果校验检测框数量不等于果实数量这是最容易翻车的地方。两个常见问题一是同一个果实被检测出多个框二是相邻果实被合并成一个框。前者需要加非极大值抑制NMSYOLOv8 默认已经做了但如果 IoU 阈值设得太高重叠框不会被抑制。后者需要看训练数据里果实间距如果标注时就把两个果实标成一个框模型学到的就是合并检测。# 调整 NMS 的 IoU 阈值 results self.model(image_path, conf0.5, iou0.45)iou参数控制 NMS 的阈值默认 0.7。调低到 0.45 会抑制更多重叠框适合果实密集且相互遮挡的场景。但调太低也可能把相邻果实的一个框误删需要根据实际图片试。另一个校验手段是设置面积过滤太小的框可能是噪声太大的框可能是误检。# 按面积过滤异常框 boxes results[0].boxes valid_count 0 for box in boxes: x1, y1, x2, y2 box.xyxy[0].tolist() area (x2 - x1) * (y2 - y1) if 100 area 50000: # 根据实际图片分辨率调整 valid_count 1面积阈值要根据图片分辨率和果实实际像素大小来定。可以先跑几张图打印出所有框的面积分布再定上下限。这一步没有固定公式靠实际数据调。注意界面显示计数结果时建议同时显示置信度分布或检测框可视化方便演示时解释为什么是这个数字。只给一个数字评委问起来不好回答。6. 避坑与常见问题排查那些让我重跑训练的血泪教训6.1 训练 loss 不下降或震荡严重现象训练开始后 box_loss 和 cls_loss 一直在高位震荡mAP 不涨。原因通常是学习率太大或数据标注有问题。先检查标注文件里有没有坐标超出 0 到 1 范围的再检查类别 id 是否从 0 开始连续。如果标注没问题把lr0降到 0.001 再跑。还有一种情况是预训练权重和模型结构不匹配比如用yolov8n.pt加载到yolov8m的配置上这时候会报错或静默失败确认 model 参数和权重文件对应。6.2 推理时检测框数量明显偏多或偏少现象同一张图界面显示的计数和肉眼数的差很多。原因可能是置信度阈值不合适或者 NMS 的 IoU 阈值需要调整。先固定一张测试图把conf从 0.1 到 0.9 逐个试看计数变化曲线选一个稳定区间。如果框大量重叠把iou从 0.7 降到 0.45 再试。另外检查输入图片的尺寸如果推理时 resize 到 640 但原图是 4000 像素宽小果实会缩到几个像素检测不到很正常。6.3 界面启动报缺少 DLL 或模块找不到现象在 Windows 上双击运行界面脚本报ImportError或缺少Qt platform plugin。原因是 PyQt5 的环境变量没配好或者虚拟环境没激活。解决方法是确保在激活的虚拟环境里运行不要直接双击 py 文件。如果还报 Qt 插件错误设置环境变量QT_QPA_PLATFORM_PLUGIN_PATH指向 PyQt5 的 plugins 目录或者重装 PyQt5pip install --force-reinstall pyqt5。6.4 训练完的权重在界面里加载失败现象训练时一切正常界面加载best.pt时报错或检测结果全空。原因是界面脚本里的模型路径写错了或者用了last.pt而不是best.pt。last.pt是最后一轮的权重可能过拟合best.pt是验证集上最好的权重优先用这个。另外确认界面脚本和训练用的 ultralytics 版本一致版本不同可能导致权重加载兼容性问题。6.5 数据集划分不合理导致验证指标虚高现象mAP 很高但实际用的时候效果差。原因是训练集和验证集里有重复图片或相似度极高的图片模型相当于在“背答案”。划分数据集时确保同一棵树的图片不要同时出现在训练集和验证集里最好按拍摄时间或地点划分。如果数据集本身不大用 8:1:1 的比例验证集至少留几十张有代表性的图。7. 进阶技巧用置信度分布和面积过滤把计数误差压到最低训练和界面都跑通之后真正决定这套系统好不好用的是计数误差。我自己的习惯是做一个简单的后处理校准先在一批测试图上跑推理把每个检测框的置信度和面积都导出来然后画散点图看分布。通常会发现置信度在 0.3 到 0.5 之间有一批模棱两可的框这些就是误差的主要来源。具体做法是分两步过滤。第一步按置信度卡一个下限比如 0.4低于这个值的框直接丢掉。第二步按面积卡上下限太小的框面积小于图片总面积的 0.1%大概率是噪声太大的框面积大于 5%大概率是误检。两个条件同时满足才计入最终数量。def calibrated_count(results, img_w, img_h, conf_min0.4, area_min_ratio0.001, area_max_ratio0.05): boxes results[0].boxes if boxes is None: return 0 img_area img_w * img_h count 0 for box in boxes: conf box.conf[0].item() x1, y1, x2, y2 box.xyxy[0].tolist() area (x2 - x1) * (y2 - y1) if conf conf_min and area_min_ratio * img_area area area_max_ratio * img_area: count 1 return count这个函数比直接len(boxes)多了一层校准参数需要根据实际图片调。conf_min从 0.4 开始试area_min_ratio和area_max_ratio根据果实占画面比例定。调好之后同一批测试图的计数误差通常能从 ±20% 压到 ±5% 以内。还有一个技巧是保存每次推理的原始结果包括框坐标、置信度和面积存成 CSV。这样后期如果发现某张图计数不对可以回溯是哪个框被误算或漏算了不用重新跑推理。我一般会在界面里加一个“导出详细结果”的按钮演示的时候如果评委问某个数字怎么来的直接打开 CSV 指给他看比口头解释有说服力。从那以后我每次做完检测类项目都会强制走一遍“置信度分布 面积过滤 结果导出”这三步不管界面看起来多正常。希望帮到你。本文还有配套的精品资源点击获取 YOLOv8果园果实自动计数毕设实战：从环境配置到界面联调
简介这份资源面向计算机、人工智能、自动化等专业的在校学生与教师提供一套基于YOLOv8的果园成熟果实自动计数完整方案可用于毕业设计、课程设计或大作业。压缩包共8个文件约15.91MB包含3个Python脚本、3个模型权重文件与2个说明文本分别对应可视化界面、模型训练与视频检测等核心环节并附有完整数据集与部署教程简单配置即可运行。项目可输出核心指标曲线图、混淆矩阵、F1分数曲线、精确率-召回率曲线、验证集预测结果及标签分布图便于答辩展示与结果分析。目前已有30人学习关注。读者可直接获得经过测试的源码、预训练权重与可视化页面既能快速复现果实计数效果也能在此基础上修改扩展功能适合作为毕设保底方案或入门目标检测的实践素材。1. 果园果实自动计数这套 YOLOv8 毕设资源到底能不能直接跑果园里数果子这件事人工干一天下来眼睛发花数错率还高。这套资源的核心就是用 YOLOv8 做成熟果实的目标检测再把检测框转成计数结果配了一个可视化界面数据集和部署教程都打包在里面。拿到手之后不需要自己从零标注、从零搭界面解压完按教程走一遍就能跑起来。适合两类人一类是毕设或课程设计时间紧、需要完整可演示系统的学生另一类是想快速验证 YOLOv8 在农业计数场景下效果的从业者。资源里包含源码、可视化界面、完整数据集和部署教程技术栈是 Python YOLOv8 图形界面框架整体走的是“检测即计数”的路线不涉及复杂的多目标跟踪。下面按实际拆包和复现的顺序把环境配置、数据组织、训练调参、界面联调和常见翻车点逐个讲清楚。2. 环境配置与依赖安装从零到能跑通第一张推理图2.1 为什么优先选 CPU 版先跑通再换 GPU很多人拿到 YOLOv8 项目第一反应是装 CUDA、配 cuDNN结果卡在驱动版本上耗掉一整天。我的建议是先用 CPU 版本把推理和界面跑通确认代码逻辑没问题再切 GPU 训练。原因很直接CPU 版安装简单不会因为显卡驱动和 PyTorch 版本不匹配导致 import 就报错。等推理通了再装 GPU 版 PyTorch 做训练这时候即使出问题你也能确定是环境问题而不是代码问题。常见做法是建一个独立虚拟环境避免和系统 Python 混在一起。Ubuntu 20.04 和 Windows 10/11 都适用下面以 conda 为例。# 创建虚拟环境Python 版本建议 3.9 或 3.10 conda create -n yolov8_fruit python3.10 -y conda activate yolov8_fruit # 安装 CPU 版 PyTorch先跑通推理用 pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu # 安装 ultralytics 和界面依赖 pip install ultralytics opencv-python pillow pyqt5这里torch和torchvision走的是 CPU 源装完大概几百 MB。ultralytics是 YOLOv8 的官方库版本建议锁在 8.0.x 到 8.1.x 之间太新的版本有时会改 API 导致项目里的推理脚本报参数错误。pyqt5是可视化界面的依赖如果项目用的是 tkinter 就换成tkinterPython 自带不用额外装。装完之后跑一句yolo predict modelyolov8n.pt sourcehttps://ultralytics.com/images/bus.jpg验证环境能输出检测结果图就说明基础环境没问题。2.2 GPU 训练环境的切换与版本对齐CPU 推理通了之后如果要训练自己的数据集GPU 是绕不开的。这时候把 CPU 版 PyTorch 卸掉换成对应 CUDA 版本的 GPU 版。关键点是先确认显卡驱动支持的 CUDA 最高版本再选 PyTorch 对应的 CUDA 版本不要反过来。# 查看显卡驱动和 CUDA 支持版本 nvidia-smi # 卸载 CPU 版安装 GPU 版以 CUDA 11.8 为例 pip uninstall torch torchvision -y pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118 # 验证 GPU 是否可用 python -c import torch; print(torch.cuda.is_available())nvidia-smi右上角显示的 CUDA Version 是驱动支持的最高版本不是当前安装的版本。比如显示 12.2你可以装 CUDA 11.8 的 PyTorch也可以装 12.1 的但不能装 12.3 的。torch.cuda.is_available()返回 True 才算成功。如果返回 False先检查驱动版本再检查 PyTorch 和 CUDA 是否匹配最后看是不是装成了 CPU 版。这一步没有后悔药版本对不上就是跑不了 GPU。注意如果项目里 requirements.txt 锁定了 torch 版本优先按它来不要自己升级。版本冲突是 YOLOv8 项目里最常见的翻车原因之一。3. 数据集组织与标注格式转换让 YOLOv8 认得你的果园图片3.1 目录结构与 data.yaml 的对应关系YOLOv8 对数据集的目录结构有固定要求不是随便放几个文件夹就能训练。标准结构是 images 和 labels 分开各自下面再分 train、val、test。资源里如果已经带了完整数据集先检查目录是不是这个结构不是的话按下面调整。fruit_dataset/ ├── images/ │ ├── train/ # 训练集图片 │ ├── val/ # 验证集图片 │ └── test/ # 测试集图片可选 ├── labels/ │ ├── train/ # 训练集标注 txt │ ├── val/ # 验证集标注 txt │ └── test/ # 测试集标注 txt └── data.yaml # 数据集配置文件data.yaml是训练时的入口配置内容必须和实际路径对应。常见写法如下path: ./fruit_dataset # 数据集根目录 train: images/train # 训练集相对路径 val: images/val # 验证集相对路径 test: images/test # 测试集相对路径没有就删掉这行 nc: 1 # 类别数果园成熟果实通常只检测一类 names: [ripe_fruit] # 类别名称顺序要和标注里的 class_id 对应nc是类别数量如果只检测成熟果实就是 1。names列表的顺序很重要标注文件里的 class_id 是 0 就对应列表第一个名字。如果数据集里还有未成熟果实或其他类别nc和names都要相应增加。路径可以用相对路径也可以用绝对路径相对路径是相对于你执行训练命令时所在的目录不是相对于 data.yaml 文件本身这一点容易搞混。3.2 标注格式转换Labelme 转 YOLO 的脚本与边界处理如果数据集里的标注是 Labelme 生成的 JSON 格式需要转成 YOLO 的 txt 格式。转换的核心是把多边形或矩形坐标归一化到 0 到 1 之间格式是class_id x_center y_center width height。下面是一个可直接用的转换脚本。import json import os from pathlib import Path def labelme_to_yolo(json_dir, output_dir, class_map): json_dir: Labelme JSON 文件目录 output_dir: 输出 YOLO txt 的目录 class_map: 类别名到 id 的映射如 {ripe_fruit: 0} json_dir Path(json_dir) output_dir Path(output_dir) output_dir.mkdir(parentsTrue, exist_okTrue) for json_file in json_dir.glob(*.json): with open(json_file, r, encodingutf-8) as f: data json.load(f) img_w data[imageWidth] img_h data[imageHeight] lines [] for shape in data[shapes]: label shape[label] if label not in class_map: continue # 跳过未定义类别避免训练时报错 class_id class_map[label] points shape[points] # 取多边形或矩形的外接框 xs [p[0] for p in points] ys [p[1] for p in points] x_min, x_max min(xs), max(xs) y_min, y_max min(ys), max(ys) # 归一化并计算中心点和宽高 x_center (x_min x_max) / 2.0 / img_w y_center (y_min y_max) / 2.0 / img_h width (x_max - x_min) / img_w height (y_max - y_min) / img_h # 边界裁剪防止坐标超出 0-1 范围 x_center min(max(x_center, 0.0), 1.0) y_center min(max(y_center, 0.0), 1.0) width min(max(width, 0.0), 1.0) height min(max(height, 0.0), 1.0) lines.append(f{class_id} {x_center:.6f} {y_center:.6f} {width:.6f} {height:.6f}) # 写入同名 txt txt_path output_dir / (json_file.stem .txt) with open(txt_path, w, encodingutf-8) as f: f.write(\n.join(lines)) if __name__ __main__: labelme_to_yolo( json_dir./raw_annotations, output_dir./fruit_dataset/labels/train, class_map{ripe_fruit: 0} )这个脚本做了三件事读取 Labelme JSON、把多边形转成外接矩形并归一化、写入 YOLO 格式 txt。class_map参数控制类别映射数据集里有什么类别就加什么没在 map 里的标注会被跳过避免训练时出现未知 class_id 导致报错。归一化后的坐标做了 0 到 1 的裁剪防止标注时点超出图片边界导致训练异常。转换完记得抽查几个 txt 文件确认坐标数值在合理范围内不要出现大于 1 或负数的情况。注意图片和标注文件名必须一一对应除了扩展名不同。如果图片是img_001.jpg标注必须是img_001.txt放在 labels 对应目录下。名字对不上训练时会被当成无标注图片跳过。4. 模型训练与参数调优从默认配置到果园场景适配4.1 训练命令与关键参数含义YOLOv8 的训练入口很简洁一条命令就能启动。但默认参数不一定适合果园果实这种密集小目标场景需要根据实际情况调整。# 基础训练命令 yolo detect train \ data./fruit_dataset/data.yaml \ modelyolov8n.pt \ epochs100 \ imgsz640 \ batch16 \ lr00.01 \ patience20 \ project./runs/train \ namefruit_expmodel参数决定用哪个预训练权重。yolov8n.pt是最小的 nano 版本速度快但精度低如果果实目标小且密集建议换成yolov8s.pt或yolov8m.pt。epochs是训练轮数100 轮起步数据集小的话可以加到 200。imgsz是输入图片尺寸640 是默认值果实占画面比例小的话可以提到 1280但显存占用会明显增加。batch是批大小显存不够就往下调从 16 降到 8 再到 4。lr0是初始学习率默认 0.01数据集小可以降到 0.001 防止过拟合。patience是早停轮数20 轮内验证指标没提升就自动停省时间。训练过程中会在runs/train/fruit_exp下生成权重文件、损失曲线和验证结果。重点关注results.csv里的metrics/mAP50和metrics/mAP50-95前者是 IoU 0.5 时的平均精度后者是 0.5 到 0.95 的平均值。果园计数场景下 mAP50 到 0.85 以上基本可用低于 0.7 就要检查数据集标注质量或增加数据量。4.2 小目标密集场景的参数调整策略果园果实往往在画面里占比较小而且成熟果实可能成簇出现默认的 640 输入尺寸和 anchor 配置不一定最优。几个实际有效的调整方向参数默认值果园场景建议原因imgsz640960 或 1280提高小目标分辨率modelyolov8nyolov8s 或 yolov8m更大模型对小目标更敏感batch168 或 4大输入尺寸下显存不够lr00.010.005 或 0.001小数据集防过拟合mosaic1.00.5 或关闭果实密集时 mosaic 可能引入干扰close_mosaic1020 或 30最后阶段关闭增强稳定收敛mosaic是 YOLOv8 默认开启的数据增强把四张图拼成一张训练。果实计数场景下如果图片本身果实就很密集mosaic 可能让模型学到错误的上下文关系适当降低比例或最后几十轮关掉。close_mosaic控制最后多少轮关闭 mosaic设大一点让模型在真实分布上多收敛几轮。训练完之后用验证集跑一遍评估命令是yolo detect val model./runs/train/fruit_exp/weights/best.pt data./fruit_dataset/data.yaml。看输出的混淆矩阵和各类别 AP如果只有一类重点看漏检和误检的比例。漏检多说明模型太保守可以降低置信度阈值误检多说明模型把背景当成了果实需要增加负样本或提高置信度阈值。5. 可视化界面联调与计数逻辑让检测框变成数字5.1 界面调用推理的代码结构与参数传递可视化界面通常是一个独立的 Python 脚本通过按钮触发图片选择、推理和结果显示。核心逻辑是加载训练好的权重对选中图片做推理把检测框画在图上同时统计框的数量作为计数结果。下面是一个典型的界面推理函数。from ultralytics import YOLO import cv2 class FruitCounter: def __init__(self, model_path, conf_threshold0.5): self.model YOLO(model_path) self.conf_threshold conf_threshold def detect_and_count(self, image_path): # 执行推理conf 控制置信度阈值 results self.model(image_path, confself.conf_threshold) # 获取检测框数量 boxes results[0].boxes count len(boxes) if boxes is not None else 0 # 绘制结果图 annotated results[0].plot() return annotated, count # 界面里调用 counter FruitCounter(./runs/train/fruit_exp/weights/best.pt, conf_threshold0.5) img, num counter.detect_and_count(./test_images/orchard_01.jpg) cv2.imwrite(./output/result_01.jpg, img) print(f检测到成熟果实数量{num})conf_threshold是置信度阈值默认 0.5。这个参数直接决定计数结果调高会减少误检但可能漏检调低会检出更多但可能把背景算进去。果园场景下建议先在验证集上试几个值看哪个阈值下计数误差最小。results[0].boxes是检测框对象len()就是数量。results[0].plot()返回画好框的图片可以直接显示在界面上或保存。界面部分如果是 PyQt5把detect_and_count绑定到按钮的点击事件上图片显示用 QLabel 的 setPixmap。如果是 tkinter用 Canvas 或 Label 显示。关键是把推理放在子线程里不要在主线程跑否则界面会卡死。这一点在演示的时候特别明显老师或评委看到界面无响应会直接扣分。5.2 计数逻辑的边界处理与结果校验检测框数量不等于果实数量这是最容易翻车的地方。两个常见问题一是同一个果实被检测出多个框二是相邻果实被合并成一个框。前者需要加非极大值抑制NMSYOLOv8 默认已经做了但如果 IoU 阈值设得太高重叠框不会被抑制。后者需要看训练数据里果实间距如果标注时就把两个果实标成一个框模型学到的就是合并检测。# 调整 NMS 的 IoU 阈值 results self.model(image_path, conf0.5, iou0.45)iou参数控制 NMS 的阈值默认 0.7。调低到 0.45 会抑制更多重叠框适合果实密集且相互遮挡的场景。但调太低也可能把相邻果实的一个框误删需要根据实际图片试。另一个校验手段是设置面积过滤太小的框可能是噪声太大的框可能是误检。# 按面积过滤异常框 boxes results[0].boxes valid_count 0 for box in boxes: x1, y1, x2, y2 box.xyxy[0].tolist() area (x2 - x1) * (y2 - y1) if 100 area 50000: # 根据实际图片分辨率调整 valid_count 1面积阈值要根据图片分辨率和果实实际像素大小来定。可以先跑几张图打印出所有框的面积分布再定上下限。这一步没有固定公式靠实际数据调。注意界面显示计数结果时建议同时显示置信度分布或检测框可视化方便演示时解释为什么是这个数字。只给一个数字评委问起来不好回答。6. 避坑与常见问题排查那些让我重跑训练的血泪教训6.1 训练 loss 不下降或震荡严重现象训练开始后 box_loss 和 cls_loss 一直在高位震荡mAP 不涨。原因通常是学习率太大或数据标注有问题。先检查标注文件里有没有坐标超出 0 到 1 范围的再检查类别 id 是否从 0 开始连续。如果标注没问题把lr0降到 0.001 再跑。还有一种情况是预训练权重和模型结构不匹配比如用yolov8n.pt加载到yolov8m的配置上这时候会报错或静默失败确认 model 参数和权重文件对应。6.2 推理时检测框数量明显偏多或偏少现象同一张图界面显示的计数和肉眼数的差很多。原因可能是置信度阈值不合适或者 NMS 的 IoU 阈值需要调整。先固定一张测试图把conf从 0.1 到 0.9 逐个试看计数变化曲线选一个稳定区间。如果框大量重叠把iou从 0.7 降到 0.45 再试。另外检查输入图片的尺寸如果推理时 resize 到 640 但原图是 4000 像素宽小果实会缩到几个像素检测不到很正常。6.3 界面启动报缺少 DLL 或模块找不到现象在 Windows 上双击运行界面脚本报ImportError或缺少Qt platform plugin。原因是 PyQt5 的环境变量没配好或者虚拟环境没激活。解决方法是确保在激活的虚拟环境里运行不要直接双击 py 文件。如果还报 Qt 插件错误设置环境变量QT_QPA_PLATFORM_PLUGIN_PATH指向 PyQt5 的 plugins 目录或者重装 PyQt5pip install --force-reinstall pyqt5。6.4 训练完的权重在界面里加载失败现象训练时一切正常界面加载best.pt时报错或检测结果全空。原因是界面脚本里的模型路径写错了或者用了last.pt而不是best.pt。last.pt是最后一轮的权重可能过拟合best.pt是验证集上最好的权重优先用这个。另外确认界面脚本和训练用的 ultralytics 版本一致版本不同可能导致权重加载兼容性问题。6.5 数据集划分不合理导致验证指标虚高现象mAP 很高但实际用的时候效果差。原因是训练集和验证集里有重复图片或相似度极高的图片模型相当于在“背答案”。划分数据集时确保同一棵树的图片不要同时出现在训练集和验证集里最好按拍摄时间或地点划分。如果数据集本身不大用 8:1:1 的比例验证集至少留几十张有代表性的图。7. 进阶技巧用置信度分布和面积过滤把计数误差压到最低训练和界面都跑通之后真正决定这套系统好不好用的是计数误差。我自己的习惯是做一个简单的后处理校准先在一批测试图上跑推理把每个检测框的置信度和面积都导出来然后画散点图看分布。通常会发现置信度在 0.3 到 0.5 之间有一批模棱两可的框这些就是误差的主要来源。具体做法是分两步过滤。第一步按置信度卡一个下限比如 0.4低于这个值的框直接丢掉。第二步按面积卡上下限太小的框面积小于图片总面积的 0.1%大概率是噪声太大的框面积大于 5%大概率是误检。两个条件同时满足才计入最终数量。def calibrated_count(results, img_w, img_h, conf_min0.4, area_min_ratio0.001, area_max_ratio0.05): boxes results[0].boxes if boxes is None: return 0 img_area img_w * img_h count 0 for box in boxes: conf box.conf[0].item() x1, y1, x2, y2 box.xyxy[0].tolist() area (x2 - x1) * (y2 - y1) if conf conf_min and area_min_ratio * img_area area area_max_ratio * img_area: count 1 return count这个函数比直接len(boxes)多了一层校准参数需要根据实际图片调。conf_min从 0.4 开始试area_min_ratio和area_max_ratio根据果实占画面比例定。调好之后同一批测试图的计数误差通常能从 ±20% 压到 ±5% 以内。还有一个技巧是保存每次推理的原始结果包括框坐标、置信度和面积存成 CSV。这样后期如果发现某张图计数不对可以回溯是哪个框被误算或漏算了不用重新跑推理。我一般会在界面里加一个“导出详细结果”的按钮演示的时候如果评委问某个数字怎么来的直接打开 CSV 指给他看比口头解释有说服力。从那以后我每次做完检测类项目都会强制走一遍“置信度分布 面积过滤 结果导出”这三步不管界面看起来多正常。希望帮到你。本文还有配套的精品资源点击获取