一种基于关键特征融合的皮带撕裂检测方法和系统

    公开(公告)号:CN118887183A

    公开(公告)日:2024-11-01

    申请号:CN202410945217.3

    申请日:2024-07-15

    摘要: 本发明公开一种基于关键特征融合的皮带撕裂检测方法和系统,包括:采集皮带表面图像构建数据集;训练双流融合网络的共享特征层提取通用特征信息,输出融合特征图供双分支任务共享;轻量化目标检测分支通过双分支感知注意力机制,增强易漏检撕裂类别特征的感知敏感度,输出定位锚框及类别预测结果和对应置信度Cd;轻量化分割决策分支通过误检特征增强模块,精准区分干扰物与真实撕裂,输出目标真正为撕裂的预测结果和对应置信度Cs;将Cd和Cs输入置信度融合器,输出被检测目标最终分类,同时根据目标检测分支的定位结果标出最终检测锚框位置;使用训练完成的双流融合网络,对被检皮带表面图像进行撕裂检测。本发明可以对皮带撕裂进行精准检测。