无人机人员救援数据集
使用DJI Phantom 4A拍摄,分辨率为1920×1080像素。涵盖了多种场景,包括碎石路上、采石场、高低草丛、森林阴影等地的人员画面,图像中人员姿势分为站立、坐着、躺着、行走、奔跑。共1981张图像6500个不同姿势的标记,2.5GB,提供voc和yolo格式标注
一、无人机人员救援数据集参数表
| 项目 | 详情 |
|---|---|
| 数据集名称 | 无人机俯视视角人员救援检测数据集 |
| 拍摄设备 | DJI Phantom 4A(大疆精灵4A) |
| 图像分辨率 | 1920×1080 像素 |
| 覆盖场景 | 碎石路面、采石矿场、高低草丛、林地阴影区、野外开阔地、临水滩涂等复杂野外救援场景 |
| 人员姿态类别 | 站立、坐立、躺卧(被困/受伤姿态)、行走、奔跑(覆盖被困人员全状态) |
| 图像总量 | 1981张 |
| 标注实例总数 | 6500个不同姿态的人员标注框 |
| 数据集大小 | 2.5GB |
| 标注格式 | 同时提供PASCAL VOC、YOLO通用格式,直接适配主流目标检测框架 |
三、YOLO格式训练代码(开箱可用)
基于YOLOv8框架编写,适配本数据集俯视视角、小目标、复杂光影的特点,包含训练、验证、推理全流程:
# 第一步:安装依赖# pip install ultralytics opencv-python tqdm# 第二步:编写数据集配置文件 person_rescue.yaml""" path: ./drone_person_rescue # 替换为你的数据集根目录 train: images/train # 训练集图片文件夹路径 val: images/val # 验证集图片文件夹路径 nc: 1 # 检测类别数(仅人员1类) names: ['person'] # 类别名称 """fromultralyticsimportYOLOif__name__=='__main__':# 加载预训练模型(算力弱选yolov8n,追求精度选yolov8m/l)model=YOLO('yolov8s.pt')# 启动训练(针对无人机俯视场景优化增强参数)train_results=model.train(data='person_rescue.yaml',epochs=100,imgsz=1920,# 匹配数据集原生分辨率,不损失小目标细节batch=8,# 16G显存建议设为8,8G显存设为4device=0,# 用GPU训练填0,CPU训练填'cpu'workers=4,project='drone_rescue_train',name='rescue_person_best',patience=20,# 早停:20轮精度不提升自动终止# 针对无人机场景优化数据增强augment=True,mosaic=1.0,# 马赛克增强提升小目标检测能力mixup=0.1,hsv_v=0.5,# 亮度增强,适配树荫、阴影场景flipud=0.5,# 上下翻转,适配无人机俯视视角fliplr=0.5,scale=0.5,)# 验证模型精度metrics=model.val()print(f"验证集mAP50:{metrics.box.map50:.4f}, mAP50-95:{metrics.box.map:.4f}")# 加载最优权重推理best_model=YOLO('./drone_rescue_train/rescue_person_best/weights/best.pt')# 单张图片推理best_model.predict('test_drone_img.jpg',save=True,conf=0.25)# 无人机实时视频流推理(替换为你的无人机RTSP地址即可)# best_model.predict('rtsp://你的无人机视频流地址', stream=True, conf=0.3, save=True)四、核心应用场景
- 野外失联搜救:搭载训练后模型的无人机可在山区、荒野自动扫描,快速识别高草丛、树荫下目视难以发现的被困人员,比人工搜救效率提升10倍以上。
- 矿山/事故救援:采石场坍塌、矿区事故后,无人机快速巡查现场,精准识别躺卧、坐立等受伤被困人员姿态,辅助救援力量精准定位。
- 森林消防/山地救援:森林、山地复杂地形下,自动识别迷路游客、受伤被困人员,解决复杂地貌下人员搜索难度大的问题。
- 灾后全域排查:地震、泥石流等自然灾害后,无人机批量扫查受灾区域,自动识别存活被困人员位置,指导救援力量优先投放。
- 野外安防巡逻:边境、自然保护区、禁入区域的无人机巡逻,自动识别违规闯入人员,支持奔跑、行走等动态人员检测。
无人机人员救援数据集 关键词标签
一、综合大类标签
无人机数据集、人员检测数据集、救援识别数据集、航拍目标检测、野外搜救数据集
二、设备&图像参数标签
DJI Phantom 4A、1080P、1920×1080、航拍图像、高清图像
三、场景环境标签
碎石路、采石场、草丛、高低植被、森林、树荫阴影、野外复杂场景、山地环境、露天矿区
四、目标姿态标签
人员姿态检测、站立、坐立、躺卧、行走、奔跑、被困人员、多姿态人体
五、标注格式标签
VOC格式、YOLO格式、目标检测标注、图像标注、边框标注
六、技术&算法标签
YOLO训练、PASCAL VOC、计算机视觉、视觉检测、小目标检测、俯视视角检测
七、行业应用标签
应急救援、野外搜救、矿山救援、森林救援、灾后搜救、安防巡逻、山地救援、防灾减灾