Skip to main navigation Skip to search Skip to main content

An Optimized SSD Target Detection Algorithm Based on K-Means Clustering

  • Yonggang Chi*
  • , Jialin Fan
  • , Bo Pang
  • , Yuelong Xia
  • *Corresponding author for this work
  • Harbin Institute of Technology

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

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.

Original languageEnglish
Title of host publicationMachine Learning and Intelligent Communications - 5th International Conference, MLICOM 2020, Proceedings
EditorsMingxiang Guan, Zhenyu Na
PublisherSpringer Science and Business Media Deutschland GmbH
Pages148-156
Number of pages9
ISBN (Print)9783030667849
DOIs
StatePublished - 2021
Event5th International Conference on Machine Learning and Intelligent Communications, MLICOM 2020 - Shenzhen, China
Duration: 26 Sep 202027 Sep 2020

Publication series

NameLecture Notes of the Institute for Computer Sciences, Social-Informatics and Telecommunications Engineering, LNICST
Volume342
ISSN (Print)1867-8211
ISSN (Electronic)1867-822X

Conference

Conference5th International Conference on Machine Learning and Intelligent Communications, MLICOM 2020
Country/TerritoryChina
CityShenzhen
Period26/09/2027/09/20

Keywords

  • Deep learning
  • K-means
  • SSD network
  • Target detection

Fingerprint

Dive into the research topics of 'An Optimized SSD Target Detection Algorithm Based on K-Means Clustering'. Together they form a unique fingerprint.

Cite this