A Shallow Neural Network for Recognition of Strip Steel Surface Defects Based on Attention Mechanism
-
- Li Dan
- School of Electrical and Information Engineering, Anhui University of Technology
-
- Ge Shiquan
- School of Electrical and Information Engineering, Anhui University of Technology
-
- Zhao Kai
- School of Electrical and Information Engineering, Anhui University of Technology
-
- Cheng Xing
- School of Electrical and Information Engineering, Anhui University of Technology
抄録
<p>This research proposes an efficient strip steel surface defect classification model (ASNet) based on convolutional neural network (CNN), which can run in real time on commonly used serial computing platforms. We only used a very shallow CNN structure to extract features of the defect images, and an attention layer which makes the model ignore some irrelevant noise and obtain an effective description of the defects is designed. In addition, a nonlinear perceptron is added to the top of the model to recognize defects based on the extracted features. On the strip steel surface defect image dataset NEU-CLS, our model achieves an average classification accuracy of 99.9%, while the number of parameters of the model is only 0.041 M and the computational complexity of the model is 98.1 M FLOPs. It can meet the requirements of real-time operation and large-scale deployment on a common serial computing platform with high recognition accuracy.</p>
収録刊行物
-
- ISIJ International
-
ISIJ International 63 (3), 525-533, 2023-03-15
一般社団法人 日本鉄鋼協会
- Tweet
詳細情報 詳細情報について
-
- CRID
- 1390013950050769152
-
- ISSN
- 13475460
- 09151559
-
- 本文言語コード
- en
-
- データソース種別
-
- JaLC
- Crossref
-
- 抄録ライセンスフラグ
- 使用不可