Real-Time Steel Surface Defect Detection and Classification with Inference Acceleration
- Title
- Real-Time Steel Surface Defect Detection and Classification with Inference Acceleration
- Authors
- Moon, Seongrok; Lee, Jun Hui; Kim, Kyung Soo; Park, Chan; Park, PooGyeon
- Date Issued
- 2023-10-18
- Publisher
- IEEE Computer Society
- Abstract
- Detecting defects on surfaces is a crucial challenge in the steel industry. Various object detection models, including one-stage and two-stage approaches, have been developed to address this problem. Currently, there is a scarcity of models that can effectively and efficiently handle both defect detection and classification tasks in real time. It is widely recognized that striking a balance between inference speed and the accuracy of object detection is a critical aspect that needs to be addressed. To address this challenge, our objective was to develop a model that ensures a high detection rate while achieving real-time processing capabilities. In pursuit of this objective, we conducted a comparative analysis between YOLOv7, a one-stage model, and Faster R-CNN, a two-stage model, followed by model optimization using TensorRT to enhance both inference speed and detection performance. As a result, we have successfully implemented a defect detection model utilizing actual production data, which achieved a detection rate of approximately 98.8% and a false ratio of 20%, while operating at a speed of 46 frames per second (FPS). This achievement demonstrates the effectiveness of our approach in balancing detection accuracy and inference speed.
- URI
- https://oasis.postech.ac.kr/handle/2014.oak/121476
- ISSN
- 1598-7833
- Article Type
- Conference
- Citation
- 23rd International Conference on Control, Automation and Systems, ICCAS 2023, page. 994 - 998, 2023-10-18
- Files in This Item:
- There are no files associated with this item.
Items in DSpace are protected by copyright, with all rights reserved, unless otherwise indicated.