YOLO电气工程图纸符号目标检测数据集
建筑空间中的目标检测:从标注到部署
📊 数据集基本信息
- 目标类别: [‘3wayswitch’, ‘AA’, ‘ACpower supply’, ‘AIR CIRCUIT BREAKER WITH SYN’, ‘AIR CIRCUIT BREAKER WITH SYN. FACILITY’, ‘AIR CIRCUIT BREAKER WITHOUT SYN’, ‘AIR CIRCUIT BREAKER WITHOUT SYN. FACILITY’, ‘AMMERTER’, ‘AMMETER SELECTOR SWITCH’, ‘AUTO MANUAL SELECTOR SWITCH’, ‘Air circuit breaker’, ‘Ammeter’, ‘Announciator’, ‘BRANCH CIRCUIT MINIMUM SIZE PER CKT BKR’, ‘BRANCH CIRCUIT MINIMUM SIZE PER CKT BKR - REMAIN’, ‘BUS BAR’, ‘Battery’, ‘Brake Cluthes’, ‘Busbar’, ‘CABLE TERMINATION’, ‘CT CURRRENT TRANSFORMER’, ‘Capacitive Voltage transformer’, ‘Circuit Breaker’, ‘Connection point’, ‘Connectors’, ‘Counter’, ‘Coupling capacitor’, ‘Current transformer’, ‘DIESEL GENERATOR’, ‘DISTRIBUTION TRANSFORMER’, ‘Disconnector Isolator’, ‘Disconnector Switch’, ‘Distribution Panel Meter’, ‘ELT’, ‘ELTR’, ‘EMERGENCY STOP PUSH BUTTON’, ‘EXISTING POWER -WHIP- CONNECTION TO CUBICLES - REMOVE’, ‘Earth ground’, ‘FIELD WIRING’, ‘Field isolator terminal’, ‘Generator’, ‘HOME RUN TO PANEL MINIMUM SIZE PER CKT BKR’, ‘HOME RUN TO PANEL MINIMUM SIZE PER CKT BKR - REMAIN’, ‘INDICATION LAMP’, ‘LED -Light Emitting Diode’, ‘LOW VOLTAGE WIRING’, ‘Light Dependent Resistor’, ‘MASTER TRIP RELAY’, ‘MCB MINIATURE CIRCUIT BREAKER’, ‘MCCB MOULDED CASE CIRCUIT BREAKER’, ‘MULTIFICTIN RELAY WITH IEC’, ‘MULTIFUNCTION METER’, ‘MUTIFUNCTION RELAY WITH IEC 61850’, ‘Motor terminal’, ‘PLC terminal’, ‘PT POTENTIAL TRANSFORMER’, ‘Potential transformer’, ‘Potentiometer’, ‘RCD Circuit Breaker’, ‘RESTRICTED EARTH FAULT RELAY’, ‘Resistor’, ‘SANDWICH TYPE BUS DUCT’, ‘SWITCH LEG - LETTER DENOTES CONNECTION ZONE’, ‘Single Phase energy meter’, ‘Switch disconnector fuse’, ‘Synchronous generator’, ‘TEST TERMINAL BLOCK’, ‘THREE-CONDUCTORS’, ‘TNC SELECTTOR SWITCH TRIP NEUTRAL CLOSE’, ‘TRIP CKT SUPERVISION RELAY’, ‘Transistor’, ‘Transistor -NPN-’, ‘VOLTMETER’, ‘VOLTMETER SELECTOR SWITCH’, ‘Wave trap’, ‘auxillarycontacts’, ‘branch’, ‘cabletray’, ‘camswitches’, ‘capacitor’, ‘circuit breaker with overload protection’, ‘circuit_breaker’, ‘connectionpoint’, ‘contactors’, ‘control suppply isolator’, ‘dc power supply’, ‘diacsandtriacs’, ‘diode’, ‘disconnector’, ‘door indicator’, ‘doublepole switch’, ‘doublepoleswitch’, ‘drawout circuit breaker’, ‘emergency switch’, ‘flow sensor’, ‘footswitch’, ‘fuse’, ‘heater’, ‘home and building’, ‘hydralic motors’, ‘hydraulic misc’, ‘hydraulic valve’, ‘hydrauliccylinder’, ‘hydraulicpump’, ‘inductor’, ‘isolation check’, ‘layout’, ‘levelsensor’, ‘light sensor’, ‘limit switch’, ‘lubrication shed’, ‘mcc drive cell terminal’, ‘meter’, ‘motion sensor’, ‘motor’, ‘octo coupler’, ‘overload trip’, ‘pandid’, ‘panel’, ‘photo electric sensor’, ‘pilot light’, ‘pneumatic misc’, ‘pneumatic motor’, ‘pneumatic valve’, ‘pneumaticcylinder’, ‘pneumaticpump’, ‘power supply’, ‘prefabricated circuit’, ‘pressure sensor’, ‘proximity sensor’, ‘pull buttons’, ‘push button’, ‘relay’, ‘resistor’, ‘selector switch’, ‘single pole switch’, ‘socket’, ‘solenoid’, ‘surge protector’, ‘switch’, ‘temperature sensor and controller’, ‘terminal block’, ‘thermistor’, ‘timer’, ‘transformer’, ‘vendor cable’, ‘voltmeter’]
- 中文类别:[‘三路开关’, ‘AA电池’, ‘交流电源’, ‘带同步装置的空气断路器’, ‘带同步设施的空气断路器’, ‘不带同步装置的空气断路器’, ‘不带同步设施的空气断路器’, ‘安培表’, ‘安培表选择开关’, ‘自动手动选择开关’, ‘空气断路器’, ‘电流表’, ‘指示器’, ‘每回路断路器分支电路最小尺寸’, ‘每回路断路器分支电路最小尺寸-剩余’, ‘母线’, ‘电池’, ‘制动离合器’, ‘母线排’, ‘电缆终端’, ‘电流互感器’, ‘电容式电压互感器’, ‘断路器’, ‘连接点’, ‘连接器’, ‘计数器’, ‘耦合电容器’, ‘电流互感器’, ‘柴油发电机’, ‘配电变压器’, ‘隔离开关’, ‘隔离开关’, ‘配电盘仪表’, ‘紧急定位发射器’, ‘紧急定位发射接收器’, ‘紧急停止按钮’, ‘现有电源-鞭状-连接至柜体-移除’, ‘接地’, ‘现场布线’, ‘现场隔离端子’, ‘发电机’, ‘至配电盘主回路每断路器最小尺寸’, ‘至配电盘主回路每断路器最小尺寸-剩余’, ‘指示灯’, ‘发光二极管’, ‘低压布线’, ‘光敏电阻’, ‘主跳闸继电器’, ‘微型断路器’, ‘塑壳断路器’, ‘符合IEC标准的多功能继电器’, ‘多功能电表’, ‘符合IEC 61850标准的多功能继电器’, ‘电机端子’, ‘PLC端子’, ‘电压互感器’, ‘电压互感器’, ‘电位器’, ‘漏电断路器’, ‘限制性接地故障继电器’, ‘电阻器’, ‘夹层式母线槽’, ‘开关支路-字母表示连接区域’, ‘单相电能表’, ‘开关熔断器组合’, ‘同步发电机’, ‘测试端子排’, ‘三导体’, ‘TNC选择开关跳闸中性闭合’, ‘跳闸回路监视继电器’, ‘晶体管’, ‘NPN型晶体管’, ‘电压表’, ‘电压表选择开关’, ‘阻波器’, ‘辅助触点’, ‘分支’, ‘电缆桥架’, ‘凸轮开关’, ‘电容器’, ‘带过载保护的断路器’, ‘断路器’, ‘连接点’, ‘接触器’, ‘控制电源隔离器’, ‘直流电源’, ‘双向二极管和双向晶闸管’, ‘二极管’, ‘隔离开关’, ‘门指示器’, ‘双极开关’, ‘双极开关’, ‘抽屉式断路器’, ‘紧急开关’, ‘流量传感器’, ‘脚踏开关’, ‘保险丝’, ‘加热器’, ‘住宅与建筑’, ‘液压马达’, ‘液压杂项’, ‘液压阀’, ‘液压缸’, ‘液压泵’, ‘电感器’, ‘隔离检查’, ‘布局’, ‘液位传感器’, ‘光传感器’, ‘限位开关’, ‘润滑棚’, ‘MCC驱动单元端子’, ‘仪表’, ‘运动传感器’, ‘电动机’, ‘八通道耦合器’, ‘过载跳闸’, ‘PANDID设备’, ‘配电盘’, ‘光电传感器’, ‘指示灯’, ‘气动杂项’, ‘气动马达’, ‘气动阀’, ‘气动缸’, ‘气动泵’, ‘电源’, ‘预制电路’, ‘压力传感器’, ‘接近传感器’, ‘拉钮’, ‘按钮’, ‘继电器’, ‘电阻器’, ‘选择开关’, ‘单极开关’, ‘插座’, ‘电磁铁’, ‘浪涌保护器’, ‘开关’, ‘温度传感器与控制器’, ‘接线端子排’, ‘热敏电阻’, ‘定时器’, ‘变压器’, ‘供应商电缆’, ‘电压表’]
- 训练集:1838 张
- 验证集:338 张
- 测试集:336 张
- 总计:2512 张
📄 data.yaml 配置信息
该数据集提供了data.yaml文件,内容如下:
train:../train/imagesval:../valid/imagestest:../test/imagesnc:147names:['3wayswitch','AA','ACpower supply','AIR CIRCUIT BREAKER WITH SYN','AIR CIRCUIT BREAKER WITH SYN. FACILITY','AIR CIRCUIT BREAKER WITHOUT SYN','AIR CIRCUIT BREAKER WITHOUT SYN. FACILITY','AMMERTER','AMMETER SELECTOR SWITCH','AUTO MANUAL SELECTOR SWITCH','Air circuit breaker','Ammeter','Announciator','BRANCH CIRCUIT MINIMUM SIZE PER CKT BKR','BRANCH CIRCUIT MINIMUM SIZE PER CKT BKR - REMAIN','BUS BAR','Battery','Brake Cluthes','Busbar','CABLE TERMINATION','CT CURRRENT TRANSFORMER','Capacitive Voltage transformer','Circuit Breaker','Connection point','Connectors','Counter','Coupling capacitor','Current transformer','DIESEL GENERATOR','DISTRIBUTION TRANSFORMER','Disconnector Isolator','Disconnector Switch','Distribution Panel Meter','ELT','ELTR','EMERGENCY STOP PUSH BUTTON','EXISTING POWER -WHIP- CONNECTION TO CUBICLES - REMOVE','Earth ground','FIELD WIRING','Field isolator terminal','Generator','HOME RUN TO PANEL MINIMUM SIZE PER CKT BKR','HOME RUN TO PANEL MINIMUM SIZE PER CKT BKR - REMAIN','INDICATION LAMP','LED -Light Emitting Diode','LOW VOLTAGE WIRING','Light Dependent Resistor','MASTER TRIP RELAY','MCB MINIATURE CIRCUIT BREAKER','MCCB MOULDED CASE CIRCUIT BREAKER','MULTIFICTIN RELAY WITH IEC','MULTIFUNCTION METER','MUTIFUNCTION RELAY WITH IEC 61850','Motor terminal','PLC terminal','PT POTENTIAL TRANSFORMER','Potential transformer','Potentiometer','RCD Circuit Breaker','RESTRICTED EARTH FAULT RELAY','Resistor','SANDWICH TYPE BUS DUCT','SWITCH LEG - LETTER DENOTES CONNECTION ZONE','Single Phase energy meter','Switch disconnector fuse','Synchronous generator','TEST TERMINAL BLOCK','THREE-CONDUCTORS','TNC SELECTTOR SWITCH TRIP NEUTRAL CLOSE','TRIP CKT SUPERVISION RELAY','Transistor','Transistor -NPN-','VOLTMETER','VOLTMETER SELECTOR SWITCH','Wave trap','auxillarycontacts','branch','cabletray','camswitches','capacitor','circuit breaker with overload protection','circuit_breaker','connectionpoint','contactors','control suppply isolator','dc power supply','diacsandtriacs','diode','disconnector','door indicator','doublepole switch','doublepoleswitch','drawout circuit breaker','emergency switch','flow sensor','footswitch','fuse','heater','home and building','hydralic motors','hydraulic misc','hydraulic valve','hydrauliccylinder','hydraulicpump','inductor','isolation check','layout','levelsensor','light sensor','limit switch','lubrication shed','mcc drive cell terminal','meter','motion sensor','motor','octo coupler','overload trip','pandid','panel','photo electric sensor','pilot light','pneumatic misc','pneumatic motor','pneumatic valve','pneumaticcylinder','pneumaticpump','power supply','prefabricated circuit','pressure sensor','proximity sensor','pull buttons','push button','relay','resistor','selector switch','single pole switch','socket','solenoid','surge protector','switch','temperature sensor and controller','terminal block','thermistor','timer','transformer','vendor cable','voltmeter']🖼️ 标注可视化
数据集下载
数据集下载:⬇️⬇️⬇️ 点击下载
数据集详细介绍
YOLO电气工程图纸符号目标检测数据集是一个专为电气工程领域设计的图像识别数据集,其核心目标是支持对电气工程图纸中各类符号的自动识别与定位。从名称和类别数量来看,该数据集适用于电力系统设计、自动化控制、建筑电气施工等场景,能够为智能图纸解析、CAD软件辅助识别、设备布局分析等应用提供基础数据支持。在实际工程中,准确识别图纸中的符号有助于提高设计效率、降低人工错误率,并为后续的自动化处理和信息提取奠定基础。
本数据集包含2512张高质量电气工程图纸图像,覆盖147种不同的电气符号类别,涵盖开关、继电器、变压器、电缆、接线端子、指示灯等多种常见电气元件。这些类别均来源于实际工程标准,具有明确的行业定义和应用场景,确保了数据集的专业性和实用性。由于电气符号种类繁多且细节复杂,该数据集在标注时需兼顾精确性和一致性,以满足深度学习模型训练的需求。
在标注规范方面,该数据集采用了严格的矩形边界框标注方式,每个符号均被精确标记,确保模型能够学习到清晰的视觉特征。同时,标注过程中遵循统一的行业标准,避免了因人为差异导致的标注误差,提升了模型的泛化能力和可靠性。这种高质量的标注体系使得该数据集在训练目标检测模型时具备较高的参考价值,尤其适合用于YOLO系列算法的优化与部署。
该数据集的典型应用方向包括电气图纸的智能识别系统、自动化设计工具的辅助模块、以及工业设备的故障诊断与维护支持。对于企业而言,可以基于该数据集开发定制化的电气图纸分析工具,提升设计与施工效率。建议在实际应用中结合具体业务流程进行模型微调,并考虑与其他工程数据源进行融合,以增强系统的实用性和准确性。
YOLO26 目标检测算法原理
如果说 YOLOv8 统一了多任务框架,那 YOLO26 做的就是极致简化推理流程——去掉 NMS、去掉 DFL,追求端到端的干净架构。
端到端无 NMS 推理:传统 YOLO 模型在推理后需要执行非极大值抑制(NMS)来去除冗余检测框,这增加了后处理延迟和部署复杂度。YOLO26 默认采用一对一(one-to-one)检测头,直接输出每张图像最多 300 个检测结果(N, 300, 6),完全省去 NMS 步骤,推理流水线大幅简化。在 CPU 上的 ONNX 推理速度相比 YOLO11n 提升高达 43%。
无 DFL 回归:YOLO26 移除了分布焦点损失(Distribution Focal Loss, DFL)结构,检测头更加轻量。边界框回归不再受限于预设的离散区间,既降低了模型复杂度又保持了同等甚至更优的回归精度。
Progressive Loss 与 STAL:训练过程中通过渐进损失(Progressive Loss)将监督信号逐步从辅助头部转移到推理阶段的主头部。小目标感知标签分配(STAL)策略特别提升小目标的正样本标签覆盖率,改善小物体检测性能。
MuSGD 混合优化器:创新性地将大语言模型常用的 Muon 优化器与 SGD 结合,实现更稳定高效的训练收敛。相比纯 SGD,MuSGD 在大 batch 训练时收敛更快且不易震荡。
模型尺度与 COCO 性能:YOLO26 提供 N/S/M/L/X 五种尺度:
| 模型 | mAP50-95 | CPU ONNX(ms) | T4 TensorRT(ms) | 参数量 |
|---|---|---|---|---|
| YOLO26n | 40.9 | 38.9 | 1.7 | 2.4M |
| YOLO26s | 48.6 | 87.2 | 2.5 | 9.5M |
| YOLO26m | 53.1 | 220.0 | 4.7 | 20.4M |
| YOLO26l | 55.0 | 286.2 | 6.2 | 24.8M |
| YOLO26x | 57.5 | 525.8 | 11.8 | 55.7M |
Nano 版本参数量仅 2.4M,CPU 推理速度比 YOLO11n 快 43%,适合边缘设备和小数据集;X 版本 mAP 达 57.5,适用于对准确率要求极高的场景。
双头架构设计:YOLO26 检测模型内置两个检测头——一对一(默认,无 NMS,输出 300 个检测框)和一对多(需 NMS,输出 8400 个候选框)。训练时一对多头作为辅助监督信号加速收敛,推理时默认使用一对一头实现端到端输出。
YOLO26 支持全部七大视觉任务:检测、实例分割、语义分割、深度估计、分类、姿态估计和定向边界框检测(OBB),一个框架覆盖从 2D 检测到 3D 感知的完整需求。
关键训练参数配置详解
别一上来就改 lr0=0.002 或 warmup_epochs=5。YOLO26 的默认参数是经过大批量实验验证的,只在明确知道要改什么的时候才动。
| 参数 | 默认值 | 本项目设置 | 含义 |
|---|---|---|---|
model | - | yolo26n.pt | 预训练权重,n/s/m/l/x 五种尺度 |
epochs | 100 | 150 | 训练总轮数 |
imgsz | 640 | 416 | 输入图像尺寸 |
batch | 16 | 64 | 批次大小,根据显存调整 |
lr0 | 0.01 | 0.02 | 初始学习率 |
lrf | 0.01 | 0.01 | 最终学习率因子(lr0 × lrf) |
momentum | 0.937 | 0.937 | SGD 动量 |
weight_decay | 0.0005 | 0.0005 | 权重衰减(L2 正则化) |
warmup_epochs | 3.0 | 3.0 | 预热轮数 |
warmup_momentum | 0.8 | 0.8 | 预热期间动量初始值 |
box | 7.5 | 7.5 | 边界框回归损失权重 |
cls | 0.5 | 0.5 | 分类损失权重 |
dfl | 1.5 | 1.5 | DFL 损失权重(YOLO26 可置 0) |
patience | 100 | -1 | 早停轮数 |
cos_lr | False | True | 余弦学习率衰减 |
close_mosaic | 10 | 10 | 最后 N 轮关闭 Mosaic 增强 |
optimizer | auto | SGD | 优化器选择 |
amp | True | True | 自动混合精度训练 |
freeze | None | 0 | 冻结前 N 层骨干网络参数 |
dropout | 0.0 | 0.1 | 分类头 Dropout 率,防过拟合 |
三个损失权重的分工:box负责《框在哪》,cls负责《框里是什么》,dfl在 YOLO26 中可设 0 因为默认检测头已去掉 DFL。大部分场景保持默认比例即可,除非有明确的偏好需求。
学习率调参经验:
- 训练初期 loss 不下降 → 调大
lr0或增加warmup_epochs - 训练后期 loss 震荡 → 降低
lr0,增大weight_decay - 验证集 loss 先降后升(过拟合)→ 增大
dropout、weight_decay或减小epochs - 小数据集(<100张)→
lr0=0.001,weight_decay=5e-4,dropout=0.2 - 大数据集(>1000张)→
lr0=0.01,weight_decay=5e-4,batch=32+
训练可视化与TensorBoard:YOLO26 默认开启 TensorBoard 日志记录,训练过程中可实时查看 loss 曲线和指标变化。在命令行执行tensorboard --logdir runs/detect/train即可打开 Web 看板,每轮训练完成后自动刷新。训练结束后 results.png 生成的六合一曲线图是最直观的诊断工具——一眼就能看出是否过拟合、是否收敛、哪个 loss 在震荡。
断点续训机制:如果训练意外中断(断电、OOM 等),不用从头开始。YOLO 每轮保存last.pt,恢复训练时指定model=last.pt并设置resume=True,学习率和优化器状态都会从断点恢复,训练曲线也和连续训练一样平滑。
模型评估指标解读
如果训出来的 mAP50 很高但 mAP50-95 很低,说明模型对定位宽容度不够——框的位置稍微偏一点就判定为错误,通常跟标注质量有关。
训练完成后,YOLO26 训练器自动输出以下指标:
mAP50(mAP@0.5):IoU 阈值为 0.5 时的平均精度均值,反映模型粗粒度定位能力。mAP50 ≥ 0.90 表示定位效果很好。
mAP50-95(mAP@0.5:0.95):IoU 从 0.5 到 0.95 共 10 个阈值上取平均。更严苛,反映精确定位能力,是更具代表性的综合评价指标。
Precision(精确率):检测结果中正确检测的比例。高 Precision 意味着虚警少。
Recall(召回率):真实目标中被成功检测到的比例。高 Recall 意味着漏检少。
F1 Score:Precision 和 Recall 的调和平均数 = 2×P×R/(P+R),综合衡量模型平衡性。
Confusion Matrix(混淆矩阵):N×N 矩阵,对角线为正确分类,非对角线为误分类,可定位类别间混淆情况。
训练曲线分析:
train/box_loss单调下降 → 边界框回归正常收敛val/box_loss先降后升 → 过拟合风险,应增大数据增强或减小模型metrics/mAP50(B)增长放缓趋于平稳 → 模型接近收敛上限- train 持续降、val 持续升 → 严重过拟合,建议减小模型或增大 dropout
指标使用场景:
- 安全生产场景(如安全帽检测)→ 优先保证 Recall,宁可多报不能漏报
- 质检场景(如缺陷检测)→ 优先保证 Precision,降低误检率减少人工复核量
- 通用检测 → 均衡优化,关注 mAP50-95 和 F1 Score
PR 曲线实战解读:每张PR_curve.png里每个类别有一条不同颜色的曲线。不用盯着每个类看——先找出最差的那条(最靠左下角的),它代表你模型的弱项。加大那个类别的数据量或检查标注质量,通常比调任何参数都见效。
YOLO 系列算法演进史
YOLO 的每次大版本更新都解决了一个具体痛点:v3 的多尺度预测解决了小目标问题、v5 的自动锚框省去了手动计算、v26 的端到端推理彻底干掉了 NMS。
YOLOv1 (2015):将检测问题建模为单一回归问题,直接在输出层预测边界框和类别概率。速度极快但定位精度较差。
YOLOv2/YOLOv3 (2016-2018):引入 anchor box 机制、特征金字塔(FPN)和多尺度训练。YOLOv3 的 Darknet-53 骨干网络成为经典架构,至今仍被广泛使用。
YOLOv5 (2020):Ultralytics 在 PyTorch 上的实现,带来了完善的训练框架、自动锚框聚类和数据增强 pipeline,大幅降低了使用门槛。
YOLOv8 (2023):统一了检测、分割、分类和关键点任务的框架,引入 C2f 模块和无锚框(anchor-free)检测头。
YOLOv9 (2024):提出 GELAN(通用高效层聚合网络)和 PGI(可编程梯度信息),同等参数量下精度提升显著。
YOLOv10/YOLOv11 (2024):YOLOv10 首次提出无 NMS 端到端检测。YOLOv11 进一步优化骨干网络和训练策略,在速度和精度之间取得新平衡。
YOLO26 (2026):集历代之大成,核心创新包括:① 端到端无 NMS 一对一检测头 ② 无 DFL 的简化回归头 ③ Progressive Loss 渐进式监督转移 ④ STAL 小目标感知标签分配 ⑤ MuSGD 混合优化器。
关键设计演进总结:
- Anchor-based → Anchor-free(v8 起)
- C3 → C2f → C3k2(骨干模块持续优化)
- 多任务统一框架(v8 起)
- NMS 后处理 → 无 NMS 端到端(v10/v26)
- 单一损失函数 → 多任务联合损失(Box + Cls + DFL,v26 可去 DFL)
YOLO26 训练步骤详解
YOLO26 的训练步骤已经被封装得很简洁了——装好 ultralytics 包,准备 data.yaml,一条命令就能启动。
环境准备:
pipinstallultralytics yolo checks python-c"import torch; print(torch.cuda.is_available())"命令行训练:
yolo detect traindata=data.yamlmodel=yolo26n.ptepochs=150imgsz=416batch=64device=0lr0=0.02patience=-1cos_lr=Trueclose_mosaic=10Python API 训练:
fromultralyticsimportYOLO model=YOLO("yolo26n.pt")results=model.train(data="data.yaml",epochs=150,imgsz=416,batch=64,device=0,lr0=0.02,patience=-1,cos_lr=True,close_mosaic=10,augment=True,amp=True,workers=8,)训练输出文件:
runs/detect/train/weights/best.pt:验证集上 mAP 最高的权重runs/detect/train/weights/last.pt:最后一轮保存的权重runs/detect/train/results.csv:每轮 loss 和指标数据runs/detect/train/confusion_matrix.png:混淆矩阵图runs/detect/train/results.png:训练曲线图runs/detect/train/val_batch*_pred.jpg:验证集预测效果图
单张推理测试:
model=YOLO("runs/detect/train/weights/best.pt")results=model("test_image.jpg")results[0].show()多 GPU 分布式训练:
yolo detect traindata=data.yamlmodel=yolo26n.ptdevice=0,1epochs=100恢复中断的训练:
yolo detect trainmodel=path/to/last.ptdata=data.yamlresume=True部署与应用建议
大规模数据集(2512 张)训练出的模型,泛化能力通常已经不错。部署的重点在于效率和稳定性,而不是继续压榨精度。
fromultralyticsimportYOLO model=YOLO("runs/detect/train/weights/best.pt")# 推荐 TensorRT FP16(精度几乎无损、速度翻倍)model.export(format="engine",device=0,half=True)# 备选 ONNX(兼容性好)model.export(format="onnx",opset=17)运维建议:生产环境部署后建议建立模型版本管理和 A/B 测试机制。新模型上线前在 10% 流量下跑 24 小时,对比核心指标(漏检率、误检率、P99 延迟)和老模型无显著劣化再全量切换。