Research Hub

대학 자원

대학 인프라와 자원을 공유해 공동 연구와 기술 활용을 지원합니다.

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.

저자 정보

이름 소속
등록된 데이터가 없습니다.