简介无人机目标检测数据集面向目标检测算法研究者与开发者适用于训练和优化YOLO、Faster R-CNN等主流模型并可用于无人机避障、低空安防等实际场景。数据集包含7000多张无人机实拍图像刻意覆盖不同尺寸和尺度的目标有助于模型学习更泛化的特征。压缩包共两万二千六百七十五个文件其中七千五百五十九个txt文件对应YOLO标签格式七千五百五十八个xml文件对应PASCAL VOC标签格式另有七千五百五十八张jpg原图整体大小约877MB。txt格式简洁高效适合直接接入YOLO系列框架xml格式包含更丰富的元数据便于Faster R-CNN、Mask R-CNN等算法解析与扩展。数据集统一围绕“drone”类别进行标注边界框和类别信息完整免去人工标注的繁琐可支撑从模型训练到部署调优的全流程。目前已有两千二百四十八人学习下载对需要高质量单类无人机数据集的科研团队或个人具有直接参考价值。1. 目标检测无人机数据集拿到 drone-data-2.zip 先想清楚用来干什么无人机航拍目标检测和常规地面监控完全不同。飞行高度把目标像素压得极小一辆车在 4K 航拍图里经常只占 20×20 像素加上俯视视角造成目标姿态和背景干扰都与常规数据差异很大直接用通用数据集训出来的模型往往漏检严重。drone-data-2.zip 这类无人机数据集的用处就是提供俯视视角下带标注的航拍样本作为目标检测模型迁移学习的起点。五年以上经验的工程师拿到 zip 后第一反应不该是急着解压训练而是先回答三个问题标注格式是什么、类别分布是否均衡、目标尺度集中在什么范围。无人机数据集里最常见的坑是标注框不贴目标、类别 ID 不连续、图像分辨率过高导致取块训练困难。这三个问题不解决后面跑出的 mAP 都只是“看起来高”。下面按处理航拍数据集的完整流程走一遍解压统计、格式转换、YOLOv8 训练评估最后落在无人机场景的三个提分技巧上。2. drone-data-2.zip 解包后先摸清目录结构、标注格式与数据分布2.1 用列表参数查看压缩包内部再决定如何解压拿到 drone-data-2.zip 先不要直接unzip到底先用列表模式看压缩包里有哪些东西unzip -l drone-data-2.zip | head -40 unzip drone-data-2.zip -d dataset/第一行只列文件不解压重点看三件事图片是否按序列分目录、标注文件是集中在 annotations 目录还是与图片混排、有没有 README 或类别说明文件。第二行把内容解压到指定目录避免小文件散落得到处都是。这类无人机数据集解压后常见组织如下dataset/ ├── images/ │ ├── seq_001/ │ └── seq_002/ ├── annotations/ │ ├── seq_001.json │ └── seq_002.json └── README.md提示如果压缩包内没有 README先打开一个标注文件确认格式不要凭文件名猜坐标含义。同一 seq 内的连续帧之间相似度很高属于同分布样本。做训练集和验证集划分时要按 seq 切分而不是按单帧随机切否则同一航线的前后帧同时出现在两边验证 mAP 会虚高换到新航线上立刻露馅。2.2 标注格式与 bbox 坐标体系的差异无人机数据集的标注格式基本逃不出三种COCO 的 JSON、Pascal VOC 的 XML、YOLO 的 TXT。三者对同一目标的框表示方式完全不同格式常见后缀bbox 表示方式COCO JSON.json[x, y, w, h]像素绝对坐标左上角原点Pascal VOC XML.xml(xmin, ymin, xmax, ymax)像素绝对坐标YOLO TXT.txt(cx, cy, w, h)中心点加宽高归一化到 0~1很多无人机数据集标注时用 COCO 或 VOC训练 YOLO 系列模型前要转成 YOLO TXT。转换时最容易搞混的就是坐标语义COCO 的 bbox 是“左上角 x、左上角 y、宽度、高度”不是右下角VOC 的 bndbox 才是“左上角与右下角两个点”。同一个框在两种格式里元素顺序完全不同转换前先拿单张图验证不要批量转完再回头排查。拿到新数据集我会随机抽两张图把标注画出来看一遍。坐标体系判断错了画出来是框和物体完全不贴类别 ID 判断错了物体上会标成完全无关的名字。这个检查比看任何说明文档都可靠。2.3 用脚本统计类别和数量暴露长尾问题解压之后我一般会写一小段 Python 看类别分布判断这个数据集值不值得做完整训练import json from collections import Counter with open(dataset/annotations/annotations.json, encodingutf-8) as f: data json.load(f) cat_id2name {c[id]: c[name] for c in data[categories]} box_count Counter(cat_id2name[a[category_id]] for a in data[annotations]) print(图片数量:, len(data[images])) print(标注框数量:, len(data[annotations])) for name, cnt in box_count.most_common(): print(f{name:12s} {cnt:6d})这段代码把 category_id 映射回类别名再用 Counter 统计每个类别的框数量。无人机数据集中“person 1200、car 5000、truck 18”这样的分布是常态。truck 这种量级的类别在训练时几乎学不出来需要单独决定是删除、合并还是补外部数据硬放在训练集里只会拉低整体精度。3. 把 drone-data-2.zip 的标注转成 YOLO 训练格式3.1 为什么要统一成 YOLO 格式YOLOv8 所在的 ultralytics 库虽然也能直接训练 COCO 标注但大多数数据增强和部署链路默认读取 YOLO TXT。更实际的原因是YOLO 格式每个 txt 文件和图片一一对应某帧数据出问题可以单独删除修改排查起来比埋在 JSON 里直观得多。转换时要确认类别 ID 是否连续。COCO JSON 的 category_id 从 1 开始且不一定连续YOLO 的类别 ID 从 0 开始且必须连续。中间缺一个 ID训练时类别数和标签数对不上日志里会报 class index 越界或者混淆矩阵整体错位。这一点在两个格式之间的差异用一个表说得更清楚对比项COCO JSONYOLO TXT组织方式一个 JSON 集中存全部标注每张图一个 txt类别 ID从 1 开始可跳跃从 0 开始必须连续坐标像素绝对坐标归一化相对坐标检查便捷度需要写脚本解析直接打开即可读3.2 一份可直接改用的 COCO 转 YOLO 脚本把 COCO JSON 转成 YOLO 格式的 txt常见做法是写一段转换函数改输入输出路径就能用于整个压缩包内的所有标注文件#!/usr/bin/env python3 # coco2yolo.py —— 将 COCO JSON 转成 YOLO 训练格式 import json import os from pathlib import Path def coco2yolo(json_path: str, out_dir: str) - None: with open(json_path, encodingutf-8) as f: coco json.load(f) images {im[id]: im for im in coco[images]} # COCO 的 category_id 重映射为连续的 YOLO class id cat_map {c[id]: idx for idx, c in enumerate(coco[categories])} os.makedirs(out_dir, exist_okTrue) lines_by_img {} for ann in coco[annotations]: img images[ann[image_id]] x, y, w, h ann[bbox] # COCO bbox: [x, y, width, height] cx (x w / 2) / img[width] # 转中心点并归一化 cy (y h / 2) / img[height] nw w / img[width] nh h / img[height] cls cat_map[ann[category_id]] lines_by_img.setdefault(ann[image_id], []).append( f{cls} {cx:.6f} {cy:.6f} {nw:.6f} {nh:.6f} ) for img_id, lines in lines_by_img.items(): txt_name Path(images[img_id][file_name]).stem .txt Path(out_dir, txt_name).write_text(\n.join(lines) \n) if __name__ __main__: coco2yolo(dataset/annotations/annotations.json, dataset/labels)最关键的是cat_map重映射COCO 的 category_id 从 1 开始且与顺序无关YOLO 要求 0 开始连续用 enumerate 建字典即可。坐标转换是把左上角加宽高的形式改成中心点加宽高再各自除以图像宽高完成归一化。输出时保留 6 位小数是为了保证精度又不让文本文件膨胀。数值越大表达同一像素位置越精细但通常到 1e-6 已经足够覆盖 10000 像素以内的长边。3.3 划分 train/val 目录并顺手核验标签转出来的 txt 文件名必须和图片名精确一致否则训练时找不到对应标签。目录按这个结构组织/work/drone-data-2/ ├── images/ │ ├── train/ │ └── val/ └── labels/ ├── train/ └── val/划分按 seq 而不是按单帧。如果压缩包里提供了官方划分文件直接用官方划分不要自己重新随机分。划分完成后花 30 秒核验坐标越界和类别超范围find labels -name *.txt | xargs awk { if ($10 || $13) print FILENAME, $0 }这里假设数据集只有 3 个类别类 ID 取值应小于等于 3。坐标越界的核验类似检查 $3、$4、$5、$6 是否落在 0 到 1 之间。无人机数据尤其要做这一步因为 4K 图在边界处的标注框很容易出现归一化后越界的情况。4. 无人机视角下的目标检测训练配置与数据增强4.1 小目标密集分布场景默认参数需要动这几项通用目标检测训练配置按常规数据集设定直接套在无人机数据上会明显掉点。影响最大的是输入分辨率YOLOv8 默认 imgsz640一张航拍图里车只有二三十像素下采样后只剩几个像素不可能检出。这类数据我一般把 imgsz 提到 1024 或 1280用显存换检出率。其他需要调整的增强参数整理成一个表参数默认值无人机场景建议调整理由imgsz6401024~1280小目标需要更高的输入分辨率degrees0.020~40俯视图无固定“上”方向旋转增强直接提升泛化flipud0.00.5上下翻转对航拍图同样有效scale0.50.3~0.6过大会让本就细小的小目标缩没close_mosaic1015最后若干轮关闭 mosaic避免小目标被切碎影响收敛逻辑在于无人机是俯视视角目标没有地面图像里常见的“正立”约束旋转和翻转成为成本最低的增强手段而 scale 需要克制因为很多目标本身只有几十像素再缩小就直接消失了。4.2 一条完整的训练命令与参数说明把上面的思考落到命令上用 YOLOv8s 做基础模型我一般这么写yolo detect train \ datadrone_data.yaml \ modelyolov8s.pt \ epochs120 \ imgsz1024 \ batch16 \ device0 \ degrees30 \ flipud0.5 \ scale0.4 \ close_mosaic15 \ projectruns/drone \ nameexp1datadrone_data.yaml指向数据集描述文件里面定义 train、val 路径和类别名列表。modelyolov8s.pt用预训练权重做迁移速度和精度平衡。imgsz1024是影响结果最大的参数显存不够先降 batch 到 8。close_mosaic15让最后 15 个 epoch 关闭 mosaic 增强避免模型在拼图形成的伪目标上过拟合。能用 GPU 就指定device0CPU 也能训练但速度慢一个数量级建议先把 imgsz 降到 640 做通断验证确认流程没问题再上全量。4.3 类别不均衡的处理尺度无人机数据集的类别长尾比常规数据集更严重。对占优的 person、car增强参数按上面的表调整即可对数量只有几十的类别任何训练技巧都救不回分布最好直接放弃或者在外部补同类数据。判断标准是数样本数最少类不足最多类的十分之一模型基本只会把这个类学成背景误检还会拉低整体精度。常见做法是把该类从 names 列表剔除重新生成 labels等拿到更多标注再合并训练。5. 用 YOLOv8 完成目标检测评估与失败样例分析5.1 在验证集上跑一次完整评估训练结束后先用保留的验证集做评估不要只盯着训练日志里的 lossyolo detect val \ datadrone_data.yaml \ modelruns/drone/exp1/weights/best.pt \ plotsTrue \ imgsz1024 \ batch16plotsTrue会额外输出混淆矩阵、PR 曲线和预测对比图这些图是判断模型哪里不行的一手材料。命令行打印的 metrics 包含 precision、recall、mAP50、mAP50-95评估时重点看 mAP50-95它对小目标更苛刻能反映框精度的把握程度。5.2 拆开目标尺寸看精度而不是只盯一个 mAP无人机数据集目标尺寸差异大单个 mAP 会把问题掩盖住。Ultralytics 输出里包含按尺寸拆分的统计项判读逻辑是Small AP 明显低于 Medium 和 Large说明小目标漏检是主要矛盾下一步在 imgsz 和测试时增强上继续加码Small 还行但 Large 崩了大概率是标注本身有问题比如大目标框不完整或类别标错。现象可能原因下一步操作small AP 低输入分辨率不够imgsz 提到 1280或对图分块训练Large AP 也低标注框与目标不贴抽样检查原图标注某类 precision 低类别容易与背景混淆降低该类权重或补充负样本误检集中在边缘切块边界问题推理时用重叠滑窗5.3 用预测图反查训练数据的标注质量航拍数据标注质量参差不齐是常态。每次训练完我都会打开 plots 输出的预测图专门找“没标全”的区域图中明显有车但标签完全没覆盖说明原始标注漏标率较高这类数据对训练是负贡献。零星几张直接从训练集移走大面积存在就要考虑这一批数据是否值得继续用作训练集。无人机数据集数量本来就少花在清洗上的时间会直接反映在最终评估指标上。6. 无人机目标检测提速与精度补偿的三个具体技巧6.1 用测试时增强补小目标漏检航拍小目标在单次推理时容易漏测试时增强是见效最快的方式。评估和推理时打开 augment模型对输入做多尺度变换后综合结果小目标检出率有提升代价是耗时翻倍yolo detect val modelruns/drone/exp1/weights/best.pt \ datadrone_data.yaml imgsz1536 augmentTrueimgsz 提到 1536 且打开 augment 后如果 Small AP 回升说明模型没到表达上限只是输入尺度不够。这种做法适合离线出结果不适合实时推理链路。6.2 对高度相似的临近帧做去重航拍视频按帧抽取时连续几帧几乎相同进入训练集后会放大同一条航线对模型权重的影响。常见做法是计算相邻帧的感知哈希相似度高于阈值只保留一帧pip install imagehash python - EOF import imagehash from PIL import Image from pathlib import Path files sorted(Path(images).glob(*.jpg)) prev None for p in files: h imagehash.average_hash(Image.open(p)) if prev is not None and abs(h - prev) 5: p.unlink() # 相似度过高删除当前帧 else: prev h EOF汉明距离小于 5 表示两帧在感知层面几乎一样。跑完再统计一遍类别分布确认删掉的是不影响分布的冗余帧。6.3 导出 TensorRT 引擎供机载推理无人机机载平台算力有限PyTorch 直接推理很难达到实时落地时常见做法是导出 TensorRT engineyolo export modelruns/drone/exp1/weights/best.pt \ formatengine device0 halfTrue imgsz640导出后模型结构固定推理时用yolo detect predict modelbest.engine sourcedrone_photo.jpg即可。导出时的 imgsz 要与实际应用匹配机载端因为算力限制通常退回 640和训练时的 1024 不一致会有精度损失实际项目要在速度与精度之间取一个中间值而不是直接沿用训练参数。本文还有配套的精品资源点击获取