改进的RetinaFace复杂构件关键点检测算法

来源：计算机应用与软件杂志2024年第8期北京时间：

作者：林鑫, 沈建新, 秦顺, 潘峰

摘要：为了解决航天装备复杂构件进行喷涂作业时难以定位、缺少关键点数据集等问题，在三维建模软件上搭建构件模型，通过截图和标记关键点制作数据集，并针对数据集量少的问题采取数据增操作。研究现有的RetinaFace关键点检测算法并进行改进，将主干特征提取网络采用优化的MobileNet结构，学习率采用余弦退火衰减，算法输入、输出张量长度与不同构件对应的关键点数相一致。实验结果表明，模型迭代500轮后在验证集上的平均误差降至0.062，能够有效地检测出待喷涂构件的关键点，性能优于同类算法。

关键词：航天设备, 复杂构件, 关键点检测, 数据集, RetinaFace, 余弦退火衰减