基于自适应深度特征编码器的手机屏幕缺陷检测方法
首发时间:2024-01-23
摘要:在本文中,我们提出了一种基于u-net网络的新型手机屏幕缺陷检测模型。由于u-net网络简单的架构和高准确性,在缺陷检测领域得到了广泛应用。然而,我们观察到传统的u-net编码器结构过于简单,仅包含少量卷积层,这限制了网络充分捕捉目标的关键特征。同时,它缺乏合理的机制来利用深层和浅层特征信息,导致分割结果在细节区域缺乏准确性。此外,u型结构底部特的征图通道数量较多,编码器和解码器之间的直接连接增加了计算复杂性,使其不适用于实时需求。为了解决这些问题,我们引入了一个动态堆叠编码器来替代传统的u-net编码器。这个动态堆叠编码器可以根据特定目标动态调整对深层和浅层特征图的利用,从而实现更准确的分割性能。另外,为了简化网络结构并降低计算复杂性,我们引入了一个路径融合模块。该模块将u-net底部的编码-解码结构融合成一个单一路径,显著减少了参数数量和推理时间。最后,我们在损失函数中引入了一个边界损失函数,并采用加权联合训练策略,平衡网络对缺陷边界和主要区域的关注,从而全面提高网络的分割准确性。
关键词:
for information in english, please click here
a mobile screen defect detection method with adaptive deep feature encoder
abstract:in this paper, we propose a novel mobile screen defect detection framework based on the u-net network. the u-net network has been widely utilized in the field of defect detection due to its simple architecture and high accuracy. however, we observed that the traditional u-net encoder structure is excessively simplistic, consisting of only a few convolutional layers, which hinders the network from adequately capturing crucial features of the target. additionally, it lacks a rational mechanism to leverage deep and shallow-level feature information, resulting in a lack of accuracy in the segmentation results, particularly in the detailed regions. moreover, the u-shaped structure's bottom feature map has a large number of channels, and the direct connection between the encoder and decoder increases the computational complexity, making it unsuitable for real-time requirements. to address these issues, we introduce a dynamic stacked encoder to replace the traditional u-net encoder. this dynamic stacked encoder can dynamically adjust the utilization of deep and shallow-level feature maps based on specific targets, thereby achieving more accurate segmentation performance. furthermore, to simplify the network structure and reduce computational complexity, we incorporate a path fusion module. this module fuses the encoding-decoding structure located at the bottom of the u-net into a single path, significantly reducing the number of parameters and inference time. finally, we introduce a boundary loss function in the loss function and adopt a weighted joint training strategy to balance the network's attention towards defect boundaries and main regions, thereby comprehensively enhancing the segmentation accuracy of the network.
keywords:
基金:
论文图表:
引用
导出参考文献
no.****
动态公开评议
共计0人参与
勘误表
基于自适应深度特征编码器的手机屏幕缺陷检测方法
评论
全部评论0/1000