首页 理论教育6.4.4 迁移学习模型的测试结果与分析

6.4.4 迁移学习模型的测试结果与分析

【摘要】:针对本章构建的迁移学习模型,根据上述实验设置的超参数值和最佳冻结层数,分别在KTH-TIPS和UMD纹理库上开展实验,对本章构建的迁移学习模型进行最终的测试,以便获得在测试集上的纹理识别精度。对UMD纹理库进行划分,其中每类的20个样本作为训练集,10个样本作为验证集,10个样本作为测试集。

针对本章构建的迁移学习模型,根据上述实验设置的超参数值和最佳冻结层数,分别在KTH-TIPS和UMD纹理库上开展实验,对本章构建的迁移学习模型进行最终的测试,以便获得在测试集上的纹理识别精度。在这里,对KTH-TIPS纹理库进行划分,其中每类的40个样本作为训练集,10个样本作为验证集,31个样本作为测试集。对UMD纹理库进行划分,其中每类的20个样本作为训练集,10个样本作为验证集,10个样本作为测试集。为了获得稳定的实验结果,在每个纹理库上独立进行10次实验,并将这10次实验结果的平均值作为最终的纹理识别结果。同时,为了进一步评估本章算法的性能,将本章算法与其他一些先进算法的识别精度进行比较。表6-2、表6-3分别展示了本章算法和其他一些先进算法在KTH-TIPS纹理库、UMD纹理库上的识别精度对比。

表6-2 不同算法在KTH-TIPS纹理库上的识别精度对比

表6-3 不同算法在UMD纹理库上的识别精度对比

从表6-2和表6-3可以看出:

(1)本章算法在KTH-TIPS纹理库上取得了99.94%的识别精度,在UMD纹理库上取得了99.91%的识别精度,一致超越了其他所有参与比较的先进算法的识别精度。

(2)和其他基于CNN模型的方法相比,其他CNN类算法都是直接利用现有的纹理图像样本对初始的CNN模型进行训练的,但现有的纹理图像训练样本的数量较少,这导致对深度CNN模型的训练不充分,最终获得的识别精度不够理想,而本章算法通过采用迁移学习方法,能够同时利用ImageNet大型数据集和现有纹理图像样本的特征,获得了更高的识别精度。

综上所述,本章算法在KTH-TIPS纹理库、UMD纹理库上都取得了最高的识别精度,这表明本章采用的基于VGG-16模型和迁移学习的方法在纹理图像识别任务中具有良好的识别能力,和传统方法相比,进一步提高了纹理图像的识别效果。