Loading...
2022
Metal Surface Defect Detection and Classification using EfficientNetV2 and YOLOv5
한국전자통신학회
논문정보
- Publisher
- 한국전자통신학회 논문지
- Issue Date
- 2022-08-31
- Keywords
- -
- Citation
- -
- Source
- -
- Journal Title
- -
- Volume
- 17
- Number
- 04
- Start Page
- 577
- End Page
- 586
- DOI
- ISSN
- 19758170
Abstract
Detection and classification of steel surface defects are?critical for product quality control in the steel industry. However, due to its low accuracy and slow speed, the traditional approach cannot be effectively used in a production line. The current, widely used algorithm (based on deeplearning) has an accuracy problem, and there are still rooms for development. This paper proposes a method of steel surface defect detection combining EfficientNetV2 for image classification and YOLOv5 as an object detector. Shorter training time and high accuracy are advantages of this model. Firstly, the image input into EfficientNetV2 model classifies defect classes and predicts probability of having defects. If the probability of having a defect is less than 0.25, the algorithm directly recognizes that the sample has no defects. Otherwise, the samples are further input into YOLOv5 to accomplish the defect detection process on the metal surface. Experiments show that proposed model has good performance on the NEU dataset with an accuracy of 98.3%. Simultaneously, the average training speed is shorter than other models.
- 전남대학교
- KCI
- 한국전자통신학회 논문지
저자 정보
| 이름 | 소속 | ||
|---|---|---|---|
| 등록된 데이터가 없습니다. | |||