Research Article
An Optimized SSD Target Detection Algorithm Based on K-Means Clustering
3 downloads
@INPROCEEDINGS{10.1007/978-3-030-66785-6_17, author={Yonggang Chi and Jialin Fan and Bo Pang and Yuelong Xia}, title={An Optimized SSD Target Detection Algorithm Based on K-Means Clustering}, proceedings={Machine Learning and Intelligent Communications. 5th International Conference, MLICOM 2020, Shenzhen, China, September 26-27, 2020, Proceedings}, proceedings_a={MLICOM}, year={2021}, month={1}, keywords={SSD network Target detection K-means Deep learning}, doi={10.1007/978-3-030-66785-6_17} }- Yonggang Chi
Jialin Fan
Bo Pang
Yuelong Xia
Year: 2021
An Optimized SSD Target Detection Algorithm Based on K-Means Clustering
MLICOM
Springer
DOI: 10.1007/978-3-030-66785-6_17
Abstract
In response to the problem that the default box size and shape of the SSD network model need to be manually set based on experience and the lack of specificity for different data, this paper uses the k-means clustering method to optimize the default box setting method of the SSD network to make the default box more consistent with the data, enhancing the self-adaptive ability of SSD default box positioning regression, thereby improving detection accuracy and detection speed. The algorithm is applied to actual aluminum defect detection, the defect detection accuracy reaches 77.6% mAP, which is 2.86% higher than the original SSD512 model, and the detection speed is increased from 37 FPS to 39 FPS.
Copyright © 2020–2025 ICST
