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.

Cette oeuvre a été publiée sous la license Creative Commons Attribution 4.0 (CC-BY 4.0). Il est autorisé de partager et adapter l'oeuvre tant que l'auteur est crédité et la license est indiquée (avec le lien ci-dessus). Vous devez aussi indiquer si des changements on été fait vis-à-vis de l'original.