无人机频射信号检测数据集：YOLOv5训练与94.3%识别率实战

简介这份无人机频射信号检测数据集面向从事无人机侦测、频谱识别与目标检测的算法工程师及高校研究者可用于训练和评估射频信号图像中的无人机目标检测模型帮助解决复杂电磁环境下无人机信号识别精度不足的问题。资源包共729个文件包含364张jpg原始图像、364个txt标注文件以及1个yaml配置文件压缩包约136.82MB标注格式兼容yolov5可直接接入主流检测框架进行训练与验证。数据集覆盖多类无人机频射信号样本平均正确识别率可达94.3%适合作为模型对比实验、算法调优与课程设计的基准数据。目前已有451人学习下载读者可据此快速搭建训练流程、复现检测效果并在此基础上开展数据增强、模型轻量化等扩展研究为无人机监测与频谱安全应用提供可靠的数据支撑。1. 无人机频射信号检测数据集364 张原图与 94.3% 识别率背后的落地价值拿到一份标注好的无人机频射信号数据集第一反应往往不是兴奋而是怀疑——364 张原始图片平均正确识别率标称 94.3%支持 YOLOv5 格式标注这几个数字放在一起到底能不能直接喂给模型跑起来我拆过不少射频信号类的数据集多数要么是纯 IQ 采样文件、要么标注格式和主流检测框架对不上光做格式转换就得耗掉半天。这份资源的思路不一样它把无人机频射信号以图像形式固化下来文件名里带着kong-和Mini3Pro_5-8GHZ这类前缀说明采集时已经按设备型号和频段做了分组标注直接对齐 YOLOv5 的 txt 格式省掉了从零写转换脚本的环节。它适合两类人一类是想快速验证射频信号目标检测可行性的算法工程师另一类是需要一个干净起点来做无人机频射信号检测课程设计或原型验证的从业者。364 张不算多但作为冷启动数据集够你把整条训练链路跑通一遍。2. 射频信号转图像检测为什么选 YOLOv5 而不是从头造轮子2.1 频射信号做成图像到底解决了什么问题射频信号本身是一维时序数据传统做法是提取频谱特征后接分类器但这种方式对信号出现的位置、持续时间、调制变化的定位能力很弱。把频射信号转成时频图常见做法是短时傅里叶变换或小波变换后映射到像素强度信号就从「一段波形」变成了「一张有纹理的图」目标检测那套锚框回归的机制就能直接复用。这份数据集里的 jpg 文件本质上就是已经完成时频变换后的图像产物你拿到手不需要再碰 SDR 采集和变换代码直接进入标注解析和模型训练环节。选 YOLOv5 而不是 YOLOv8 或更晚的版本理由很实际YOLOv5 的工程成熟度最高配置文件结构清晰data.yaml里改几个路径就能跑对新手最友好。而且这份数据集的标注格式就是按 YOLOv5 的规范来的——每张图对应一个同名 txt每行是class_id x_center y_center width height坐标全部归一化到 0 到 1 之间。你如果硬要上 YOLOv8格式虽然兼容但配置文件字段名有差异反而多一层折腾。常见做法是先用 YOLOv5 把基线跑出来确认数据本身没有标注错位或类别混乱再考虑迁移到其他框架。2.2 数据集目录结构与标注格式核对拿到压缩包解压后你看到的应该是一堆 jpg 和一个标注文件夹。文件名里kong-29-_jpg.rf.9cd17aec...这种长哈希是采集系统自动生成的唯一标识kong-前缀大概率对应某一类无人机型号或信号源Mini3Pro_5-8GHZ则直接标明了设备型号和频段。标注文件需要和图片同名同目录或者按 YOLOv5 惯例放在labels/下、图片放在images/下。我一般会先跑一段脚本核对图片和标注是否一一对应避免训练时出现「有图无标」或「有标无图」的静默失败。import os from pathlib import Path img_dir Path(images) lbl_dir Path(labels) img_stems {p.stem for p in img_dir.glob(*.jpg)} lbl_stems {p.stem for p in lbl_dir.glob(*.txt)} # 找出缺失标注的图片和缺失图片的标注 missing_labels img_stems - lbl_stems missing_images lbl_stems - img_stems print(f图片总数: {len(img_stems)}) print(f标注总数: {len(lbl_stems)}) print(f缺标注的图片: {missing_labels}) print(f缺图片的标注: {missing_images})这段脚本的逻辑很直白分别收集图片和标注文件的 stem不含扩展名的文件名做集合差。参数上唯一需要注意的是img_dir和lbl_dir的路径要按你实际解压后的结构改。如果输出里missing_labels非空说明那些图片没有对应标注训练时 YOLOv5 会直接跳过它们不会报错但你的有效样本数就少了。missing_images非空则更危险YOLOv5 在构建数据集缓存时可能直接抛异常。我习惯在正式训练前把这两个集合都清空确保 364 张图全部有标。2.3 从零配置 YOLOv5 训练环境环境配置这块血泪经验是别用最新版的 PyTorch 去配老版 YOLOv5版本错配会导致torch.load报奇怪的错。稳妥做法是建一个干净的 conda 环境Python 3.8 或 3.9PyTorch 1.12 左右然后克隆 YOLOv5 仓库到本地。conda create -n drone_rf python3.9 -y conda activate drone_rf pip install torch1.12.1 torchvision0.13.1 --index-url https://download.pytorch.org/whl/cu113 git clone https://github.com/ultralytics/yolov5.git cd yolov5 pip install -r requirements.txtcu113对应 CUDA 11.3如果你机器上是 CUDA 11.6 或 11.8把 index-url 里的版本号改掉即可。装完后用python -c import torch; print(torch.cuda.is_available())确认 GPU 能被识别。这一步翻车最多的情况是显卡驱动太旧torch.cuda.is_available()返回 False训练会退到 CPU 上364 张图虽然不多但 CPU 训练一轮也要十几分钟调参效率极低。2.4 编写 data.yaml 与类别映射YOLOv5 训练的核心配置文件是data.yaml它告诉框架去哪里找训练集、验证集以及类别名称和数量。这份数据集如果只有一个类别无人机频射信号nc就是 1names写[drone_rf]。如果你按文件名前缀把kong-和Mini3Pro分成两类那nc改成 2names对应写两个。我一般先按单类跑通确认 mAP 正常后再考虑细分。# data.yaml train: ../dataset/images/train val: ../dataset/images/val nc: 1 names: [drone_rf]路径用相对路径或绝对路径都行但train和val指向的目录里必须同时有images和labels子目录或者图片和标注混放。YOLOv5 默认会去train路径的同级找labels如果结构不对它会报「No labels found」。我通常把 364 张图按 8:2 随机划分训练集 291 张验证集 73 张划分脚本用sklearn.model_selection.train_test_split或者手写随机抽样都行关键是固定随机种子保证每次划分一致。3. 训练参数怎么设从 364 张图里榨出 94.3% 的识别率3.1 小样本下的超参选择与迁移学习策略364 张图在目标检测里属于小样本从头训练必然过拟合。标准做法是加载 YOLOv5 的预训练权重用 COCO 上学到的通用特征来初始化主干网络然后只微调检测头。命令里--weights yolov5s.pt就是干这个的。yolov5s是最小的模型参数量少在小数据集上反而不容易过拟合别一上来就用yolov5x。python train.py \ --img 640 \ --batch 16 \ --epochs 100 \ --data data.yaml \ --weights yolov5s.pt \ --cache--img 640是输入分辨率如果你的时频图里信号细节很细可以提到 1280但显存占用会翻倍。--batch 16在 8GB 显存上比较稳显存小就降到 8。--epochs 100对小数据集够了再多容易过拟合。--cache把图片缓存到内存364 张图占不了多少内存但能明显加快每个 epoch 的读取速度。我一般还会加--patience 20如果 20 个 epoch 验证集指标不升就自动停省得干等。3.2 训练过程监控与 mAP 解读训练启动后终端会打印每个 epoch 的损失和 mAP。重点看mAP0.5它表示 IoU 阈值 0.5 时的平均精度。标称的 94.3% 正确识别率大概率对应的是这个指标或者分类准确率。如果训练到 50 个 epoch 时mAP0.5还在 0.6 左右晃说明要么标注有问题要么学习率太大。YOLOv5 默认用余弦退火学习率初始lr0是 0.01小数据集上可以降到 0.001改--lr0 0.001即可。python train.py \ --img 640 \ --batch 16 \ --epochs 100 \ --data data.yaml \ --weights yolov5s.pt \ --lr0 0.001 \ --patience 20另一个容易忽略的参数是--rect它做矩形推理对长宽比差异大的时频图能减少填充带来的无效计算。如果你的图片都是正方形加不加无所谓如果宽高比差异大加上它。3.3 验证集评估与混淆矩阵排查训练结束后用val.py在验证集上跑一遍输出详细的 precision、recall 和 mAP。python val.py \ --data data.yaml \ --weights runs/train/exp/weights/best.pt \ --img 640 \ --conf 0.25--conf 0.25是置信度阈值低于这个值的检测框会被过滤掉。如果验证集上 recall 很低但 precision 很高说明模型太保守把--conf降到 0.1 试试。反过来 precision 低 recall 高说明误检多把--conf提到 0.4 以上。YOLOv5 还会在runs/val/exp/下生成混淆矩阵和 PR 曲线混淆矩阵能直接看出有没有类别混淆——如果单类数据集出现大量背景误检多半是标注框画得太松把信号周围的噪声也框进去了。3.4 推理测试与单张图片验证训练完的模型用detect.py做推理拿几张没参与训练的图看看实际效果。python detect.py \ --weights runs/train/exp/weights/best.pt \ --source ../dataset/images/val \ --img 640 \ --conf 0.3 \ --save-txt--save-txt会把检测结果按 YOLO 格式存下来方便你对比预测框和真实标注的偏差。--source可以指向单张图、整个文件夹甚至视频文件。我习惯先拿验证集里 mAP 最低的那几张图单独跑看看模型到底在哪些样本上翻车。常见情况是信号持续时间短的样本漏检这时候要么增加这类样本的权重要么在数据增强里加时间掩蔽。4. 避坑与排查射频信号数据集训练中最容易翻车的五个点4.1 图片和标注不同名导致静默跳过现象训练日志里train的图片数量比实际文件数少但没有任何报错。原因YOLOv5 按文件名 stem 匹配图片和标注如果图片叫kong-29-_jpg.rf.9cd17aec.jpg标注却叫kong-29-_jpg.rf.9cd17aec.txtstem 一致没问题但如果标注文件多了个后缀或者少了哈希段匹配就失败。解决用 2.2 节的脚本核对确保每个 jpg 都有同名 txt不一致的手动重命名或写脚本批量改。4.2 标注坐标未归一化导致框全错现象训练 loss 从一开始就很高降不下去mAP 始终接近 0。原因YOLO 格式要求坐标是归一化到 0 到 1 的相对值如果标注里存的是绝对像素坐标比如x_center320模型学到的就是错误的位置映射。解决读几个标注文件看数值是否都在 0 到 1 之间。如果超出用图片宽高做除法批量归一化。import cv2 from pathlib import Path img_dir Path(images) lbl_dir Path(labels) for lbl_path in lbl_dir.glob(*.txt): img_path img_dir / (lbl_path.stem .jpg) if not img_path.exists(): continue h, w cv2.imread(str(img_path)).shape[:2] lines lbl_path.read_text().strip().splitlines() new_lines [] for line in lines: cls, x, y, bw, bh map(float, line.split()) # 如果坐标大于 1说明是绝对坐标需要归一化 if x 1 or y 1 or bw 1 or bh 1: x, y, bw, bh x/w, y/h, bw/w, bh/h new_lines.append(f{int(cls)} {x:.6f} {y:.6f} {bw:.6f} {bh:.6f}) lbl_path.write_text(\n.join(new_lines))这段脚本会遍历所有标注发现坐标大于 1 就按图片宽高归一化。注意cv2.imread返回的 shape 是(height, width, channels)别把 h 和 w 搞反了。4.3 训练集和验证集分布不一致现象训练集 mAP 很高验证集 mAP 低一大截。原因随机划分时把同一设备、同一频段的图片全分到了训练集验证集里全是没见过的型号模型泛化不过来。解决按文件名前缀做分层抽样保证kong-和Mini3Pro在训练集和验证集里都有。如果数据量实在少可以用 K 折交叉验证但 364 张图做 5 折每折验证集才 70 多张指标波动会很大建议至少跑 3 次不同随机种子取平均。4.4 显存溢出导致训练中断现象训练到一半报CUDA out of memory。原因--img设太大或者--batch设太大显存不够。解决先把--batch降到 8 或 4如果还不行就降--img到 416 或 320。另一个技巧是加--accumulate 2用梯度累积模拟大 batch显存占用不变但等效 batch size 翻倍。4.5 推理时置信度阈值设错导致漏检现象验证集 mAP 正常但实际推理时很多信号没框出来。原因detect.py默认--conf 0.25如果模型输出的置信度普遍偏低这个阈值就把真阳性也滤掉了。解决先用--conf 0.05跑一遍看模型到底输出了多少框再根据 PR 曲线选一个平衡点。我一般会在验证集上扫一遍 0.1 到 0.5 的阈值选 F1 最高的那个。5. 进阶技巧用 TTA 和模型集成把 94.3% 再往上推一推单模型跑通之后如果想把识别率从 94.3% 再往上提最省事的办法是测试时增强TTA。YOLOv5 的val.py和detect.py都支持--augment参数它会对每张图做水平翻转、缩放等多尺度推理然后把结果融合。代价是推理时间翻几倍但 mAP 通常能涨 1 到 2 个百分点。python val.py \ --data data.yaml \ --weights runs/train/exp/weights/best.pt \ --img 640 \ --augment \ --conf 0.25另一个技巧是模型集成用不同的随机种子训练 3 个yolov5s推理时把三个模型的检测框做加权框融合WBF。YOLOv5 仓库里自带ensemble.py但需要你先把每个模型的推理结果存成 txt再用 WBF 脚本合并。我试过在类似规模的射频数据集上三个模型集成能把 mAP0.5 从 0.94 推到 0.96 左右但代价是维护三套权重和推理管线值不值得看你的精度要求。还有一个容易被忽略的点是输入分辨率。时频图里信号细节的尺度差异很大--img 640可能把弱信号压没了。我一般会跑两组实验一组 640一组 1280看验证集 mAP 哪个高。1280 的显存占用大约是 640 的四倍8GB 卡上--batch得降到 4 才能跑。如果 1280 比 640 高不到 1 个百分点就别折腾了推理速度划不来。最后说一个我踩过的坑数据增强里的mosaic对射频时频图不一定友好。mosaic会把四张图拼成一张如果四张图的信号强度差异大拼接边界会出现明显的强度突变模型可能学到这种伪影而不是信号本身。我一般会在训练后期关掉mosaic用--close-mosaic 10让最后 10 个 epoch 不用 mosaic让模型在干净数据上收尾。从那以后我每次跑小样本检测任务都会先确认标注归一化、再核对训练验证分布、最后才调超参这三步走完基本不会出大岔子。希望帮到你。本文还有配套的精品资源点击获取