基于相似一致性自蒸馏的无人机巡检缺陷分类方法及装置
Abstract:
本发明公开一种基于相似一致性自蒸馏的无人机巡检缺陷分类方法及装置,设计了一个自蒸馏缺陷图像分类模型,采用自蒸馏训练策略压缩模型以便于在小型应用设备上实现高精度图像分类工作,该模型主要分为两部分:基于相似一致性知识构建和相似一致性知识传递。基于相似一致性知识构建部分通过计算Mini‑batch内实例间的相关关系,得到相似矩阵。相似一致性知识传递部分则在自蒸馏缺陷图像分类模型层间传递相似矩阵,细化底层次相似性,捕获丰富的上下文场景和局部特征信息。本发明针对图像分类模型容量大且复杂,无法在小型无人载具设备上使用的问题,设计了自蒸馏模型,提高缺陷图像的分类效率和精度。
Patent Agency Ranking
0/0