The development of industrial automation and intelligent manufacturing has promoted the research of computer vision and machine learning in the automated detection of surface defects in metal and plastic products, to solve the problems of low efficiency and high inconsistency in traditional manual detection. In response to such demandsthis paper proposes a lightweight model that can be used for high-precision industrial defect detection. Data Preprocessing First, we use Python to write programs to read images. Secondly, image features are extracted. In this paper, the HOG algorithm is used to extract the features of the image, and the principal component analysis (PCA) is used to reduce the dimension of the feature. Finally, we trained a defect target detection model based on grid division and LightGBM, which we call the Grid-HOG-PCA-LightGBM model. The recognition accuracy of the model can reach 98%, and it only takes up 4000B memory size. Our models not only enable automated inspection, increasing inspection efficiency and accuracy; they also reduce costs in the production process by enabling continuous and efficient inspection.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    Surface Defect Detection Based on Machine Learning and Data Augmentation


    Contributors:
    Liu, Hengxu (author) / Xu, Yuqing (author) / Huang, Jiwang (author) / Bao, Qiaozhi (author) / Wen, Zhenhua (author)


    Publication date :

    2024-10-23


    Size :

    797592 byte




    Type of media :

    Conference paper


    Type of material :

    Electronic Resource


    Language :

    English



    Wheel Defect Detection With Machine Learning

    Krummenacher, Gabriel / Ong, Cheng Soon / Koller, Stefan et al. | IEEE | 2018

    Free access

    Rail surface defect detection based on deep learning

    Li, Xiaoqing / Zhou, Ying / Chen, Hu | SPIE | 2020


    UAV Payload Detection Using Deep Learning and Data Augmentation

    Ku, Ilmun / Roh, Seungyeon / Kim, Gyeongyeong et al. | IEEE | 2022


    LANGUAGE-BASED OBJECT DETECTION AND DATA AUGMENTATION

    SCHULTER SAMUEL / AICH ABHISHEK / GOPALKRISHNA VIJAY KUMAR BAIKAMPADY | European Patent Office | 2025

    Free access

    Model-Based Data Augmentation to Improve the Performance of Machine-learning Diagnostic Systems

    Kahlen, Jannis N. / Würde, Andre / Andres, Michael et al. | TIBKAT | 2022