计算机应用与软件2024，Vol.41Issue(8)：311-318,8.DOI:10.3969/j.issn.1000-386x.2024.08.045
改进的RetinaFace复杂构件关键点检测算法
A KEYPOINTS DETECTION ALGORITHM FOR COMPLEX COMPONENTS BASED ON IMPROVED RETINAFACE
摘要
Abstract
In order to improve the problems of difficulty in positioning of spraying operation,and lack of datasets in keypoints detection of complex aerospace components,models were built in the 3 D modeling software to make datasets of keypoints detection by taking screenshots and marking keypoints.Data augmentation methods were used in order to solve the problem of small sample size of datasets.On the basis of researching and improving existing RetinaFace keypoints detection algorithm,an optimized MobileNet structure was designed for the backbone feature extraction network and the learning rate was decayed by cosine warmup.The length of the input and output tensor was consistent with the number of keypoints corresponding to different components.The experimental results show that the average error on the validation set drops to 0.062 after 500 iterations of the model.The algorithm has better performance than similar algorithms,and can effectively identify the keypoints of components to be sprayed.关键词
航天设备/复杂构件/关键点检测/数据集/RetinaFace/余弦退火衰减Key words
Aerospace equipments/Complex components/Keypoints detection/Datasets/RetinaFace/Cosine decay with warmup分类
信息技术与安全科学引用本文复制引用
林鑫,沈建新,秦顺,潘峰..改进的RetinaFace复杂构件关键点检测算法[J].计算机应用与软件,2024,41(8):311-318,8.基金项目
国防重大项目培育基金项目(NP2020420). （NP2020420）
