基于双阶段优化与多级特征增强的端子排图纸检测方法

本发明公开了一种基于双阶段优化与多级特征增强的端子排图纸检测方法，包括利用切片技术对图纸进行切片并通过基于规则的交互式合成数据增强方法对切片图像进行二次处理，获得增强后的切片图像数据集；将处理后的图像数据集输入多级特征增强的目标检测模型中进行检测，获得模型预测的类别与坐标结果；将预测结果与真实结果输入到双焦点损失函数中进行计算，得到损失值，并通过反向传播优化模型，经过多轮迭代训练出最终检测模型；将检测图纸经过切片的图像输入到训练好的模型中，并对切片结果进行融合，得到最终检测结果；本发明有效地处理了图纸中类别不平衡的问题，并通过图元检测模型，显著提高了对图纸中，尤其是小目标的识别准确性。