智慧无人机巡检-6类无人机目标检测数据集该数据集包括6个类别类别分别是0垂直起降型国定翼无人机1 普通回定翼无人机2 无尾翼型固定翼无人机3 巡飞弹4 战斗机固定翼无人机5 长机翼固定翼无人机共计图片11514张数据集是txt格式训练集图片8059张验证集2302张、测试集1153张相关YOLOv8/YOLOv9/YOLOv10/YOLOv11/yolo12/RTDETR模型可直接使用6类无人机目标检测数据集表格 全YOLO系列训练代码完全按照你的数据描述整理表格清晰、代码可直接运行、适配YOLOv8~YOLO12、RTDETR一、数据集详细信息表项目详细内容数据集名称6类无人机目标检测数据集总图片数量11514 张标注格式TXT 格式YOLO标准格式类别总数6 类训练集8059 张验证集2302 张测试集1153 张支持模型YOLOv8 / YOLOv9 / YOLOv10 / YOLOv11 / YOLO12 / RTDETR任务类型无人机目标检测、低空安防、无人机识别二、6类无人机类别对照表类别ID类别名称0垂直起降型固定翼无人机1普通固定翼无人机2无尾翼型固定翼无人机3巡飞弹4战斗机固定翼无人机5长机翼固定翼无人机三、数据集配置文件drone6.yamltrain:./images/trainval:./images/valtest:./images/testnc:6names:0:垂直起降型固定翼无人机1:普通固定翼无人机2:无尾翼型固定翼无人机3:巡飞弹4:战斗机固定翼无人机5:长机翼固定翼无人机四、YOLOv8 训练代码可直接运行train_drone6.pyfromultralyticsimportYOLOimporttorch# 配置参数 DATA_YAMLdrone6.yamlMODELyolov8s.pt# 可换yolov9s.pt / yolov10s.pt / yolov11s.pt / yolov12s.ptEPOCHS120BATCH16IMGSZ640DEVICE0iftorch.cuda.is_available()elsecpu# 开始训练 modelYOLO(MODEL)model.train(dataDATA_YAML,epochsEPOCHS,batchBATCH,imgszIMGSZ,deviceDEVICE,workers8,projectdrone6_detect,nameyolo_drone6,exist_okTrue,patience20,optimizerAdamW,lr00.001,augmentTrue,mixup0.1,mosaic1.0,close_mosaic10,saveTrue,save_period10)print(✅ 6类无人机数据集训练完成)print(最优模型drone6_detect/yolo_drone6/weights/best.pt)五、推理/检测代码detect_drone6.pyfromultralyticsimportYOLO# 加载模型modelYOLO(drone6_detect/yolo_drone6/weights/best.pt)# 检测图片/视频/文件夹/摄像头resultsmodel.predict(sourcetest.jpg,conf0.25,iou0.45,saveTrue,showTrue,imgsz640)# 输出结果forrinresults:print(f检测目标数量{len(r.boxes)})六、模型评估代码val_drone6.pyfromultralyticsimportYOLO modelYOLO(drone6_detect/yolo_drone6/weights/best.pt)metricsmodel.val(datadrone6.yaml,imgsz640,batch16,conf0.001,iou0.6,plotsTrue)print(*60)print( 6类无人机检测模型评估指标)print(fmAP0.5:{metrics.box.map50:.4f})print(fmAP0.5:0.95:{metrics.box.map:.4f})print(f精确率:{metrics.box.precision.mean():.4f})print(f召回率:{metrics.box.recall.mean():.4f})print(*60)七、RTDETR 训练代码可选train_rtdetr.pyfromultralyticsimportYOLO modelYOLO(rtdetr-l.pt)model.train(datadrone6.yaml,epochs120,batch8,imgsz640,device0,projectdrone6_rtdetr)运行说明数据集结构images/trainimages/valimages/testlabels/...新建drone6.yaml复制配置运行train_drone6.py开始训练训练完用detect_drone6.py测试图片/视频
