Enhancing Bolt Object Detection via AIGC-Driven Data Augmentation for Automated Construction Inspection

In the engineering domain, the detection of damage in high-strength bolts is critical for ensuring the safe and reliable operation of equipment. Traditional manual inspection methods are not only inefficient but also susceptible to human error. This paper proposes an automated bolt damage identification method leveraging AIGC (Artificial Intelligence Generated Content) technology and object detection algorithms. Specifically, we introduce the application of AIGC in image generation, focusing on the Stable Diffusion model. Given that the quality of bolt images generated directly by the Stable Diffusion model is suboptimal, we employ the LoRA fine-tuning technique to enhance the model, thereby generating a high-quality dataset of bolt images. This dataset is then used to train the YOLO (You Only Look Once) object detection algorithm, demonstrating significant improvements in both accuracy and recall for bolt damage recognition. Experimental results show that the LoRA fine-tuned Stable Diffusion model significantly enhances the performance of the YOLO algorithm, providing an efficient and accurate solution for automated bolt damage detection. Future work will concentrate on further optimizing the model to improve its robustness and real-time performance, thereby better meeting the demands of practical industrial applications.

This creative work has been published under the Creative Commons Attribution 4.0 International (CC-BY 4.0) license which allows copying, and redistribution as well as adaptation of the original work provided appropriate credit is given to the original author and the conditions of the license are met.