[发明专利]一种基于ViT网络的图像成像设备识别方法

摘要

本发明公开了一种基于ViT网络的图像成像设备识别方法，至少包括以下步骤：预先训练得到图像成像设备识别模型；将待识别图像输入图像成像设备识别模型并输出识别结果。在模型训练中，本发明首先根据相机拍摄的校色板和原始校色板的真值获得出当前环境条件下的相机色彩校正矩阵；其次，使用视觉ViT网络进行色彩校正矩阵特征的提取，将第一步提取出的相机色彩校正矩阵作为网络的收敛目标与网络提取的色彩校正矩阵使用损失距离的计算，经过训练得到可以提取色彩校正矩阵特征的ViT网络。训练完成得到性能优异的ViT网络，当新的图像送入网络时，训练好的ViT网络便能够提取该图像的色彩校正矩阵特征向量，结合KNN分类算法得到最终的图像相机设备型号预测类别。