一种基于豪斯多夫距离计算距离损失的方法及装置

摘要

本发明涉及豪斯多夫距离技术领域，公开了一种基于豪斯多夫距离计算距离损失的方法，包括：构建基于豪斯多夫距离的网络模型，网络模型包括从前到后依次连接的基于通道注意力模块的ShuffleNetV2网络、RepBlockv2模块和基于豪斯多夫距离设计的无锚点检测头网络；将待测工业产品的数据集和训练集输入到基于通道注意力模块的ShuffleNetV2网络获取不同尺寸的提取特征图；将提取特征图输入到RepBlockv2模块，使用多分支结构配合深度可分离卷积和残差连接，进行多级预测获取融合特征图；训练后的基于豪斯多夫距离的网络模型检测待测工业产品。本发明设计豪斯多夫距离损失函数，解决了目标检测中常用损失函数与评测指标不一致的问题，并且使得预测检测框的坐标回归范围的边界更加准确和平滑。