arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2404.10664cs.CVcs.LGeess.IV

评估基于CNN自编码器的图像去噪对图像分类任务的影响

Assessing The Impact of CNN Auto Encoder-Based Image Denoising on Image Classification Tasks

Bu-Ali Sina University(布阿里·西拿大学)

Mohsen Hami, Mahdi JameBozorg

AI总结:

针对工业铸件含噪图像缺陷检测问题,提出结合噪声类型识别与适配去噪的分类流水线,显著提升了VGG16、InceptionV3等模型的缺陷检测准确率。

摘要:

现实世界中采集的图像往往受到多种类型噪声的影响,这会显著降低计算机视觉系统的性能以及视觉数据的质量。本研究提出了一种针对铸件产品含噪图像的缺陷检测新方法,研究对象聚焦于潜水泵叶轮。该方法的技术路径是在空域和频域中运用VGG16、InceptionV3等深度学习模型,识别噪声类型与缺陷状态。研究流程首先对图像进行预处理,随后针对特定噪声类别应用适配的去噪技术,目标是将噪声检测与去噪环节整合到分类流水线中,以此提升缺陷检测的准确率与鲁棒性。\n本研究在频域下采用VGG16进行噪声类型分类取得了优异结果,准确率超过99%。去噪效果方面,椒盐噪声去除后的平均结构相似性指数(SSIM)为87.9,高斯噪声去除后的平均SSIM为64.0,周期性噪声去除后的平均SSIM为81.6。这套完整方案证明了深度自编码器(AutoEncoder)模型与中值滤波作为去噪策略在实际工业场景中的有效性。最后,本研究报告显示,与既有方法相比,缺陷检测的二分类准确率得到了显著提升:使用VGG16分类器时,准确率从94.6%提升至97.0%,验证了所提噪声检测与去噪方法的效果;使用InceptionV3分类器时,准确率从84.7%提升至90.0%,进一步证实了将噪声分析融入分类流水线的优势。

英文摘要:

Images captured from the real world are often affected by different types of noise, which can significantly impact the performance of Computer Vision systems and the quality of visual data. This study presents a novel approach for defect detection in casting product noisy images, specifically focusing on submersible pump impellers. The methodology involves utilizing deep learning models such as VGG16, InceptionV3, and other models in both the spatial and frequency domains to identify noise types and defect status. The research process begins with preprocessing images, followed by applying denoising techniques tailored to specific noise categories. The goal is to enhance the accuracy and robustness of defect detection by integrating noise detection and denoising into the classification pipeline. The study achieved remarkable results using VGG16 for noise type classification in the frequency domain, achieving an accuracy of over 99%. Removal of salt and pepper noise resulted in an average SSIM of 87.9, while Gaussian noise removal had an average SSIM of 64.0, and periodic noise removal yielded an average SSIM of 81.6. This comprehensive approach showcases the effectiveness of the deep AutoEncoder model and median filter, for denoising strategies in real-world industrial applications. Finally, our study reports significant improvements in binary classification accuracy for defect detection compared to previous methods. For the VGG16 classifier, accuracy increased from 94.6% to 97.0%, demonstrating the effectiveness of the proposed noise detection and denoising approach. Similarly, for the InceptionV3 classifier, accuracy improved from 84.7% to 90.0%, further validating the benefits of integrating noise analysis into the classification pipeline.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑