改进YOLOv7的光学镜片灰尘与点状缺陷检测

Improved YOLOv7 for dust and point defects detection of optical lens

  • 摘要: 在光学镜片生产与检测过程中无法避免灰尘附着,其在外观上与麻点和砂目缺陷极为相似,但却不会对镜片性能产生实质影响。针对这一问题,设计了一套光学镜片表面缺陷采集装置,可采集高对比度图像,并改进YOLOv7网络来实现检测分类。首先,在特征提取模块中使用深度可分离卷积,并改进空间金字塔池化结构,减少参数量并扩展感受野;其次,在骨干网络中添加多通道融合特征增强模块,能够在提升相似特征识别准确率的同时,添加通道间信息交互能力;最后,修改损失函数为NWD_SIoU,提升网络对小目标的关注度。实验结果表明:最终的网络检测精度均值达到了92.9%,模型权重减少了11.8 MB,推理速度达到了62.4 f·s−1,表明该方法能够快速有效地区分灰尘与麻点及砂目缺陷。

     

    Abstract: During the production and detection process of optical lens, it is inevitable that dust will adhere to the surface. Its appearance is similar to hole and pockmark defects, however it does not have a substantial impact on the performance of the lens. In order to solve this problem, a set of optical lens surface defect acquisition device was designed to collect high-contrast images and use an improved You Only Look Once version 7 (YOLOv7) network to achieve detection and classification. Firstly, the enhanced network incorporated depthwise separable convolutions within feature extraction module and improved the spatial pyramid pooling structure to decrease the parameter count and broaden the receptive area. Secondly, when integrated multi-channel fusion feature enhancement module into the backbone network, this module could enhance the accuracy of similar feature recognition and add the ability of information interaction between channels. Finally, the loss function was modified to normalized Gaussian Wasserstein Distance_Smoothed Intersection over Union (NWD_SIoU) to improve the network's attention to small targets. The experimental results show that the mean average precision for the final network's detection capabilities attains 92.9%, alongside a reduction of 11.8 MB in the model's weight and the frames per second reaches 62.4 f·s−1, which illustrates that the method can distinguish dust from pitting and sand defects quickly and effectively.

     

/

返回文章
返回