-
公开(公告)号:CN116977696A
公开(公告)日:2023-10-31
申请号:CN202310415076.X
申请日:2023-04-17
Applicant: 襄阳湖北工业大学产业研究院
IPC: G06V10/764 , G06V10/774 , G06V10/82 , G06V10/80 , G06V10/40 , G06N3/0464 , G06N3/045 , G06N3/096
Abstract: 本发明提供一种高空作业安全带检测方法、装置、电子设备及存储介质,该方法包括:基于高空电力作业图像数据集得到训练集;基于训练集对轻量化网络模型进行训练,并基于通道蒸馏和空间蒸馏的方法将教师网络的特征信息传递给学生网络,得到图像识别模型;将待测高空电力作业图像输入至图像识别模型,得到多个预测框,从多个预测框中去除重复预测框,得到待测高空电力作业图像对应的安全带检测结果;其中,轻量化网络模型包括主干特征提取网络和多尺度特征融合网络,主干特征提取网络包括多个IB模块,特征融合网络是由卷积层、上采样层、下采样层和特征融合层构成的特征金字塔结构。本发明可以解决现有技术中安全带检测精度较低的技术问题。