The invention discloses a wood water-based paint surface defect detection method based on
computer vision, and belongs to the technical field of wood paint surface detection, and the method comprises the following steps: S1, constructing a wood paint surface defect
data set including four typical defects of bubbles, cracks, holes and scratches; s2, introducing a
wavelet convolution module to improve the
Backbone network, enhancing the image feature capture capability, and avoiding the excessive parameterization problem of a traditional method; s2.1, a
wavelet convolution module (WTConv) is added into a
Basic Block of the ResNet network, and a
feature extraction unit is improved, so that the capability of the network for capturing paint surface defect features is enhanced, and meanwhile, the calculation complexity is reduced; and S2.2, the
wavelet convolution module enhances the multi-scale feature representation of the image through
wavelet transform. And S3, designing an AIFI module, and optimizing a
feature fusion effect through a multi-scale capture
mechanism based on K and V feature information. According to the method, the multi-scale feature capturing capability of the model is enhanced, and the detection precision of defects of different types and sizes is improved.