针对传统表面缺陷检测无法适应工业复杂背景等问题,提出一种基于特征金字塔匹配和自监督的表面缺陷检测算法。首先,将两个基于通道注意力的残差网络提取的特征构成金字塔,根据网络各层输出的差异找到缺陷。其次,网络预训练的方式上采用了自我引导潜能(BYOL)自监督学习,经过自监督学习的网络可以提取通用特征,并提高缺陷检测方法的泛化性。最后,在遇到模糊图像时,采用基于不同分辨率的蒸馏训练来让学生网络充分学会提取图像的深度特征。对所提算法在3个数据集上进行了测试,实验结果证明,所提方法好于对照组,具有更高的缺陷检测精度。
图像处理 缺陷检测 特征金字塔 自监督学习 知识蒸馏 预训练网络 激光与光电子学进展
2023, 60(4): 0415006
1 浙江大学电气工程学院, 浙江 杭州 310027
2 杭州电子科技大学计算机应用技术研究所, 浙江 杭州 310018
针对基于高光谱数据的悬浮物浓度反演问题,提出一种利用预训练神经网络(PNN)进行有监督波段选择的方法,并使用随机森林和神经网络建立悬浮物浓度反演模型。 PNN算法需进行多次重复实验以获得低噪声且充分的波段重要性表达,在每次实验时选取适当数量的波段作为神经网络输入数据的特征,训练一个神经网络,并利用最后一期训练时第一层权重的L1范数、L2范数和ReLU(Rectified Linear Unit)函数来表示波段的重要性。实验结果表明,相比其他常用的波段选择算法,使用L1范数、L2范数的PNN算法能够得到信息量更大的波段集合,且其用于悬浮物浓度反演时,能够获得更优的效果。
大气光学 高光谱数据 波段选择 预训练神经网络 悬浮物浓度反演 激光与光电子学进展
2021, 58(20): 2001001
西北大学信息科学与技术学院, 陕西 西安 710127
在脊椎CT图像分割问题中,由于脊椎与组织对比度过低和噪声的影响,传统分割算法存在分割精度差和自动化程度低等问题。基于此,提出一种通过AttentionNet定位脊椎,然后使用改进的DenseUnet进行脊椎CT分割的方法。首先,对所有脊椎CT样本数据进行裁剪、重采样、灰度值归一化等预处理操作;再次,对样本使用AttentionNet训练得到具有位置信息的Attention图;然后,对传统DenseUnet进行改进,在每个Dense block加入Shuffle操作来增加网络的鲁棒性,在每个Dense block后加入1×1卷积,以降低通道数,减少网络参数量;接着使用改进后的DenseUnet对训练样本进行预训练,得到具有先验信息的预测图;最后,将Attention图、预测图及原始图像融合为三通道的训练样本作为输入,采用改进的DenseUnet训练分割模型,并在测试集上进行验证,最终实现脊椎CT自动分割。实验结果表明,所提方法的分割精度优于传统DenseUnet,是一种有效的脊椎CT自动分割方法。
图像处理 分割 Attention图 参数量 预训练 多通道融合 激光与光电子学进展
2020, 57(20): 201008
设计了一个浅层卷积神经网络来代替预训练模型中的全连接层, 将预训练网络提取的CNN特征作为图像输入设计好的浅层CNN网络, 对比微调预训练模型的方法, 能够更好地适应航拍图像定位任务。为进一步提高航拍图像的定位准确率, 利用无人机航拍图像时间连续的特点, 通过在CNN的分类阶段加入Bi-LSTM网络, 使网络在分类时能够以多张图像特征作为判断依据。实验表明, 时序图像定位方法定位准确率稳定在0. 89左右, 对比单张图像定位方法准确率提升5%左右。
无人机 航拍图像 图像分类 图像定位 预训练网络 Unmanned Aerial Vehicle (UAV) aerial image image classification image localization pre-training network CNN CNN