简介面向目标检测入门与农业智能化应用杂草与作物数据集覆盖了常见田间场景支持yolov5、yolov7、yolov8、yolov9、yolov10、yolo11等主流yolo系列算法的训练与验证。数据集已完成训练/验证/测试划分并附带data.yaml配置文件下载后可直接用于模型训练、验证与测试标注内容同时提供yolo格式txt与voc格式xml两种版本。yolo格式包含类别索引、归一化中心点坐标与宽高信息可直接用于yolo系列模型训练voc格式的xml文件保留目标类别、边界框左上角与右下角坐标等关键信息便于在需要voc格式的脚本中直接调用与转换。资源共2000个文件以xml标注文件为主压缩包大小约137.26MB内部目录结构清晰可快速定位所需数据。目前已有102人学习适合需要快速获取带标签农业图像数据的算法工程师、科研人员或相关专业学习者能省去自行采集与标注的时间减少数据预处理成本集中精力进行模型调优与应用落地。1. 为什么杂草检测数据集是农业视觉里的硬骨头做农田视觉的同行都清楚作物检测和杂草检测完全是两个难度等级。作物长得整齐、颜色统一、行距可预期而杂草是典型的「不规则目标」——同一块地里可能有十几种杂草叶片形状互相重叠颜色跟作物幼苗高度接近再加上光照变化、土壤背景干扰模型稍微偷懒就会把草当成苗。这就是为什么2722张图像带标签的杂草-作物数据集会这么抢手它直接切中了智能除草、精准喷药、农田机器人视觉这几个落地场景最缺的原料。我拿到这份数据集时的第一反应不是训练而是先搞清楚里面的标签到底长什么样。很多下载来的农业数据集图片是JPG、标注是XML也就是VOC格式而YOLO系列训练要的是TXT格式的归一化坐标。格式不转后面一切都白搭。这篇笔记就把我从解压、体检、格式转换、训练到踩坑的完整链路写清楚适合刚拿到标注数据、准备用YOLOv8或YOLOv5训练自己检测模型的从业者参考。2. 解压与数据体检拿到zip后的第一件事2.1 解压不要双击解压用命令行保目录结构这类数据集在Windows上双击解压通常没问题但如果你在服务器上跑训练Linux环境下的解压就是基本功了。常见做法是# 解压到指定目录保留压缩包内的目录层级 unzip 杂草-作物.zip -d weed_crop_dataset # 解压后先看目录结构 find weed_crop_dataset -type d | head -20 find weed_crop_dataset -type f | wc -l-d参数指定解压目标目录避免把文件散落到当前目录find统计文件总数是为了核对是否与标题里说的2722张一致。实际经验是有些压缩包里混入了重复文件或缩略图缓存文件数对不上是常有的事。2.2 标签体检VOC和YOLO格式的识别方法解压后先别急着训练花五分钟看清楚标签格式和类别定义。这份数据集是VOC格式的每个XML对应一张JPG里面记录了每个目标的类别名和bounding box坐标。# 看任意一个XML标签的内容 cat weed_crop_dataset/Annotations/weed_001.xml | head -50重点看两个地方object标签里的name是什么bndbox里的xmin、ymin、xmax、ymax是不是都在图像尺寸范围内。常见情况是类别名不统一——有的数据集里weed写成Weedcrop写成CROP后面转格式时就会出问题。2.3 数据分布检查类别不平衡直接决定训练策略2722张图听起来不少但具体到每个类别的目标数量才是关键。我一般会用脚本快速统计import os import xml.etree.ElementTree as ET xml_dir weed_crop_dataset/Annotations category_counter {} for xml_file in os.listdir(xml_dir): if not xml_file.endswith(.xml): continue tree ET.parse(os.path.join(xml_dir, xml_file)) for obj in tree.getroot().findall(object): name obj.find(name).text category_counter[name] category_counter.get(name, 0) 1 print(category_counter)这段代码用标准库的 ElementTree 遍历所有XML统计每个类别出现的总次数不需要安装额外依赖。如果发现杂草类别数量远小于作物可以提前规划是否用数据增强、是否调整损失函数里的类别权重——这比训练到一半才发现学习率怎么调都不收敛要省事得多。3. 用Python把VOC标签批量转成YOLO训练格式3.1 为什么必须转格式YOLO要的是归一化坐标YOLOv5、YOLOv8这些模型训练时读取的标注是TXT文件每行一个目标格式是class_id x_center y_center width height四个坐标值都归一化到0~1之间。而这份数据集提供的是XML里的绝对像素坐标。不转格式直接丢给训练脚本轻则报错读不到标签重则模型把整张图当背景训练成废品。3.2 转换脚本一张图对应一个TXT的标准做法import os import xml.etree.ElementTree as ET from pathlib import Path def convert_voc_to_yolo(xml_file, output_dir, class_names): tree ET.parse(xml_file) root tree.getroot() img_width int(root.find(size/width).text) img_height int(root.find(size/height).text) image_name Path(xml_file).stem txt_path os.path.join(output_dir, image_name .txt) with open(txt_path, w) as f: for obj in root.findall(object): class_name obj.find(name).text if class_name not in class_names: continue class_id class_names.index(class_name) bbox obj.find(bndbox) xmin float(bbox.find(xmin).text) ymin float(bbox.find(ymin).text) xmax float(bbox.find(xmax).text) ymax float(bbox.find(ymax).text) x_center ((xmin xmax) / 2) / img_width y_center ((ymin ymax) / 2) / img_height w (xmax - xmin) / img_width h (ymax - ymin) / img_height f.write(f{class_id} {x_center:.6f} {y_center:.6f} {w:.6f} {h:.6f}\n) class_names [weed, crop] xml_dir weed_crop_dataset/Annotations label_dir weed_crop_dataset/labels os.makedirs(label_dir, exist_okTrue) for xml_file in os.listdir(xml_dir): if xml_file.endswith(.xml): convert_voc_to_yolo(os.path.join(xml_dir, xml_file), label_dir, class_names)class_names列表的顺序决定了每个类别对应的IDweed是0、crop是1这个顺序后面训练配置里必须保持一致。坐标转换的核心是先把绝对坐标转成中心点加宽高的形式再除以图像宽高做归一化。注意我在找size/width时用了嵌套路径的写法这在XML结构不标准时更不容易出错。3.3 转换后必须做的验证反向画框检查转换完不能只看文件生成没生成要随机抽几张图把TXT标注画回去看框是不是贴合目标。这是最直接的后悔药。import cv2 import os def draw_yolo_boxes(image_path, label_path): img cv2.imread(image_path) h, w img.shape[:2] with open(label_path, r) as f: for line in f.readlines(): class_id, x_center, y_center, bw, bh map(float, line.split()) x1 int((x_center - bw / 2) * w) y1 int((y_center - bh / 2) * h) x2 int((x_center bw / 2) * w) y2 int((y_center bh / 2) * h) color (0, 255, 0) if class_id 1 else (0, 0, 255) cv2.rectangle(img, (x1, y1), (x2, y2), color, 2) cv2.imwrite(check_ os.path.basename(image_path), img) draw_yolo_boxes(weed_crop_dataset/images/weed_001.jpg, weed_crop_dataset/labels/weed_001.txt)这段代码用OpenCV把归一化坐标还原成像素坐标绿色框是作物红色框是杂草。如果框明显偏移或者大小不对多半是转换时把xmin、xmax的顺序搞错了或者图像的宽高读取有误。这一步花不了两分钟但从源头上避免了训练一个废模型的风险。3.4 目录结构按YOLO官方约定组织数据转换完成后把图片和标签按YOLO的约定整理好方便直接用YOLOv8的命令行训练不需要写自定义的数据加载器。# 标准目录结构 weed_crop_dataset/ ├── images/ │ ├── train/ │ └── val/ ├── labels/ │ ├── train/ │ └── val/ └── data.yaml划分训练集和验证集时建议按9:1或8:2随机切分但要确保同一个场景的多张连续帧不跨集合——否则模型在验证集上的表现会被严重高估部署到地里就露馅。4. YOLOv8训练自己的杂草-作物数据集关键参数与完整流程4.1 环境准备数据文件与YAML配置训练要用的data.yaml把所有关键信息汇总在一个文件里指向图片和标签的路径并定义类别名字。# data.yaml path: ./weed_crop_dataset train: images/train val: images/val nc: 2 names: [weed, crop]path是相对于当前工作目录的根路径train和val是相对路径nc是类别数量names必须与转换脚本里的class_names顺序一致。一个常见的翻车点是路径配置错误导致训练时找不到图片报错还能看懂但还有一个更隐蔽的问题——如果把path写错YOLO不会报错只会把所有图片当作背景来训练损失函数不下降这时候大部分人第一反应是调学习率实际上是路径配错了。4.2 训练命令从YOLOv8最小可跑命令开始的参数解读yolo detect train \ modelyolov8s.pt \ datadata.yaml \ epochs100 \ imgsz640 \ batch16 \ projectweed_crop_project \ nameexp_weed \ patience20model选择yolov8s.pt是在小数据量下精度和速度的折中选择。2722张图说多不多说少不少用yolov8n可能欠拟合用yolov8l或yolov8x又容易过拟合中间档位最稳。imgsz640是农业目标检测的默认尺寸杂草本身很小如果图片里目标密集可以试试imgsz960但要接受训练时间和显存翻倍。patience20表示连续20个epoch验证集指标不提升就早停防止在验证集上过拟合。4.3 损失函数与超参数调整贴地气的调参思路YOLOv8的损失函数由三部分组成——分类损失、边框回归损失和置信度损失默认权重分配在大多数数据集上表现均衡。但在杂草作物场景里小目标占比高我一般会把hsv_h、hsv_s这类数据增强参数适度调低因为农业图像的真实色彩分布很重要增强过于激进会让模型学到不存在的颜色变化。yolo detect train \ modelyolov8s.pt \ datadata.yaml \ epochs100 \ imgsz640 \ batch16 \ hsv_h0.01 \ hsv_s0.5 \ hsv_v0.3 \ scale0.3 \ fliplr0.5fliplr0.5是水平翻转的概率这个可以保留因为农田里的杂草方向本来就是随机的。scale0.3是图像缩放增强的范围我调小了一些防止把小目标缩到不可识别的程度。4.4 混合精度与显存控制穷人版训练技巧如果显卡显存不够可以加一行ampTrue启用混合精度训练大部分情况下精度损失可以忽略但显存占用几乎减半。还有一个更极端的做法是把batch调小到4甚至2配合梯度累积yolo detect train \ modelyolov8s.pt \ datadata.yaml \ epochs100 \ imgsz640 \ batch4 \ ampTrue \ cacheTruecacheTrue会把图片提前加载到内存里省去每个epoch读取磁盘的时间。如果你的服务器内存足够大这一步能让训练速度快上不少。5. 杂草检测训练的五个高频坑现象、原因与解决方案5.1 训练loss不下降查了半天发现数据集路径配错现象loss在最初的几个epoch后就不怎么变了验证集的mAP始终在0附近徘徊。原因data.yaml里的path字段写的是相对路径但当前工作目录不对或者train、val路径写成了绝对路径但指向了空目录。YOLO在这种情况下不会直接报错而是把空目录当成了空的训练集模型没有任何有效标注可以学习。解决在训练前先跑一下yolo detect train --data data.yaml --epochs 1看输出里读取了多少张图片如果数量是0赶紧回去检查路径。还有一种排查方法是写个Python脚本打印出所有图片路径和标签路径逐一确认存在性。5.2 背景误报严重模型把土壤当杂草、把杂草当作物现象训练完在测试图上跑框出了一大堆背景中的绿色阴影或者作物和杂草框全部重叠根本分不开。原因农业图像里杂草与作物的视觉特征高度重叠尤其是在幼苗期。如果数据标注本身存在边界框不贴合目标的问题——比如框得过大把背景包进去了——模型就会学到目标周围一圈土壤也是目标的一部分推理时自然到处误报。解决先用本文3.3节的反向画框验证脚本把所有训练图片抽查一遍发现边界框明显偏移的XML单独修正。如果只是少数图像有问题直接删除这些样本比手工微调边界框更省力。另外可以尝试在数据增强里增加mosaic0.5的混合概率让模型见过更多背景变化。5.3 小目标检测完全失效杂草太小根本检测不到现象训练正常mAP在验证集上看着不错但一到实际农田里拍的照片上远处的杂草一个都检测不出来。原因YOLO的下采样倍数通常是32倍输入640×640的图像特征图只有20×20一个只有10×10像素的杂草目标在下采样后就剩不到1个像素了。验证集里的图片通常和训练集同分布目标大小相似所以验证指标欺骗了你。解决真实验证必须用无人机或农机在不同高度拍摄的、未经筛选的原始图片。训练侧可以设置imgsz960或1280让输入分辨率变大同时用yolov8m或yolov8l这种特征提取能力更强的模型。另一个有效做法是给数据集的图片做切片——把大图切成若干512×512或640×640的patch小目标在patch里就变成了大目标。5.4 打标工具格式坑LabelImg导出的XML里类别名带空格或中文现象转格式脚本运行顺利但训练时YOLO报错说类别索引超出范围。原因有些标注工具在标注时允许类别名带空格比如weed crop这种或者用户手动输入了中文名如杂草。class_names列表里是英文的weedXML里是带空格的weed 字符串匹配不上class_id 自然就找错了。解决在转换脚本里加一行.strip()方法清洗类别名同时转换前打印所有出现过的类别名集合人工看一下有什么意外值。这个习惯能省下不少排查时间。5.5 过拟合2722张图训出了100%验证精度一到新场景就报废现象训练集的loss降得很低验证集的mAP到了95%以上但换了块地的图片去测效果惨不忍睹。原因2722张图本质上是同一时间、同一地块、同一光照条件下采集的分布非常窄。模型看似学到了杂草的特征实则是记住了这片地的土壤纹理和光照模式。验证集是从同一个分布里随机切的所以验证指标很好但模型没有泛化能力。解决严格按场景来做数据划分保证同一块地或同一天拍的图全部落在训练集验证集和测试集用完全独立的地块。如果数据来源本身就很单一泛化能力只能靠采集更多数据解决。数据增强能缓解但治标不治本。6. 用难例挖掘把杂草模型的精度再往上推一步训练完第一版模型后一个非常有效的习惯是收集所有推理置信度低于某阈值但仍包含真实目标的检测框把它们作为难例追加到训练集里。这个方法在杂草场景下尤其好用——地里总有一些光照极差或者叶片重叠严重的角落这些样本在原始数据集里天然稀少。具体的做法是用训练好的模型跑一遍所有训练图片筛选出IoU大于0.5但置信度低于0.4的检测框把对应的图片裁出来看标注是否准确。如果标注没问题说明模型在这个样本上确实学不到位把这类图多复制几份并在训练时提高其采样权重二次训练时模型会对这些困难场景更敏感。一轮难例挖掘完再跑一轮难例挖掘两轮之后精度会明显提升。还有一个容易被忽略的验证技巧是把模型跑在视频帧上而不是单张图片上观察同一株杂草在连续帧里的检测框是否稳定。如果框忽大忽小或者时有时无说明模型是在碰运气此时需要加大帧间上下文信息或做简单的跟踪后处理而不是继续无脑堆训练轮数。检测模型在农业场景里没有完全的银弹数据分布永远是最核心的决定因素——标注质量过关、分布覆盖到位、格式转换不出错YOLO的训练过程其实非常顺滑。这套流程我跑过不止一次最深的体会是掉进坑里时先怀疑数据再怀疑模型。希望帮到你。本文还有配套的精品资源点击获取
