1.一种行人重识别模型训练方法,其特征在于,包括:
基于预训练的两个神经网络构建第一行人重识别网络模型和第二行人重识别网络模型,每个行人重识别网络模型包括卷积神经网络、Transformer模块和全局子值池化模块;
将目标域图像集分别输入所述第一行人重识别网络模型和第二行人重识别网络模型中,利用所述每个行人重识别网络模型中的所述卷积神经网络进行处理,利用所述Transformer模块提取全局信息,得到中间特征;
所述利用所述每个行人重识别网络模型中的所述卷积神经网络进行处理,利用所述Transformer模块提取全局信息,得到中间特征包括:将所述目标域图像集输入所述每个行人重识别网络模型中,利用所述卷积神经网络进行处理,并进行空间维度展平得到特征f,将所述特征f附上cls标记得到特征fcls,并对所述特征fcls进行位置编码得到特征fcp;
将所述特征fcp输入所述Transformer模块中,利用所述Transformer模块进行处理并进行转置和维度扩展,得到所述中间特征F;
将所述中间特征输入所述全局子值池化模块中,利用所述全局子值池化模块进行池化处理,输出最终特征值,并进行聚类分析,得到所述每个行人重识别网络模型的硬伪标签;
所述将所述中间特征输入所述全局子值池化模块中,利用所述全局子值池化模块进行池化处理,输出最终特征值,并进行聚类分析,得到所述每个行人重识别网络模型的硬伪标签包括:将所述中间特征F输入所述全局子值池化模块中,提取所述全局子值池化模块任一通道的特征Fc,将所述特征Fc中的n个值进行降序排序,选取其中最大的k个值,进行加权求和得到中间特征重复上述步骤得到所述全局子值池化模块中所有通道的中间特征 将所述所有通道的中间特征 进行拼接得到所述最终特征 利用所述最终特征 进行聚类,生成所述硬伪标签;
其中 ∑pi≡1,1≤k≤n,{s1,s2,...,sn}为降序排序结果;
构建所述每个行人重识别网络模型的时序平均模型,将所述目标域图像集分别输入每个时序平均模型中,得到每个时序平均模型软伪标签;
利用所述硬伪标签和所述软伪标签对所述第一行人重识别网络模型和第二行人重识别网络模型进行协同训练;
选取两个时序平均模型中mAP精度值大的模型作为目标行人重识别网络模型。
2.如权利要求1所述的方法,其特征在于,所述Transformer模块衍生于vision transformer模型中的模块结构。
3.如权利要求1所述的方法,其特征在于,所述构建所述每个行人重识别网络模型的时序平均模型,将所述目标域图像集分别输入每个时序平均模型中,得到所述每个时序平均模型软伪标签包括:(T) (T‑1)
根据E [Θ]=αE [Θ]+(1‑α)Θ分别构建第一时序平均模型和第二时序平均模型;
将所述目标域图像集分别输入所述第一时序平均模型和所述第二时序平均模型中进行分类,得到第一软伪标签和第二软伪标签;
其中,Θ为行人重识别网络模型的参数,E[Θ]为时序平均模型的参数,T为整个目标域(0)阶段的迭代次数且E [Θ]=Θ,α为范围在[0,1)之间的超参数。
4.如权利要求3所述的方法,其特征在于,所述利用所述硬伪标签和所述软伪标签对所述第一行人重识别网络模型和第二行人重识别网络模型进行协同训练包括:S61:利用所述第一行人重识别网络模型的软硬伪标签与所述第二行人重识别网络模型的预测结果计算损失,优化所述第一行人重识别网络模型;
S62:利用所述第二行人重识别网络模型的软硬伪标签与所述第一行人重识别网络模型的预测结果计算损失,优化所述第二行人重识别网络模型;
S63:重复步骤S61和步骤S62,直至达到预设训练阈值;
S64:判断所述第一时序平均模型和所述第二时序平均模型的mAP精度值,选取时序平均模型mAP精度高的模型作为所述目标行人重识别网络模型。
5.一种行人重识别模型训练装置,其特征在于,包括:
构建模型模块,用于基于预训练的两个神经网络构建第一行人重识别网络模型和第二行人重识别网络模型,每个行人重识别网络模型包括卷积神经网络、Transformer模块和全局子值池化模块;
提取特征模块,用于将目标域图像集分别输入所述第一行人重识别网络模型和第二行人重识别网络模型中,利用所述每个行人重识别网络模型中的所述卷积神经网络进行处理,利用所述Transformer模块提取全局信息,得到中间特征;
所述利用所述每个行人重识别网络模型中的所述卷积神经网络进行处理,利用所述Transformer模块提取全局信息,得到中间特征包括:将所述目标域图像集输入所述每个行人重识别网络模型中,利用所述卷积神经网络进行处理,并进行空间维度展平得到特征f,将所述特征f附上cls标记得到特征fcls,并对所述特征fcls进行位置编码得到特征fcp;
将所述特征fcp输入所述Transformer模块中,利用所述Transformer模块进行处理并进行转置和维度扩展,得到所述中间特征F;
计算模块,用于将所述中间特征输入所述全局子值池化模块中,利用所述全局子值池化模块进行池化处理,输出最终特征值,并进行聚类分析,得到所述每个行人重识别网络模型的硬伪标签;
所述将所述中间特征输入所述全局子值池化模块中,利用所述全局子值池化模块进行池化处理,输出最终特征值,并进行聚类分析,得到所述每个行人重识别网络模型的硬伪标签包括:将所述中间特征F输入所述全局子值池化模块中,提取所述全局子值池化模块任一通道的特征Fc,将所述特征Fc中的n个值进行降序排序,选取其中最大的k个值,进行加权求和得到中间特征重复上述步骤得到所述全局子值池化模块中所有通道的中间特征 将所述所有通道的中间特征 进行拼接得到所述最终特征 利用所述最终特征 进行聚类,生成所述硬伪标签;
其中 ∑pi≡1,1≤k≤n,{s1,s2,...,sn}为降序排序结果;
生成软标签模块,用于构建所述每个行人重识别网络模型的时序平均模型,将所述目标域图像集分别输入每个时序平均模型中,得到每个时序平均模型软伪标签;
训练优化模块,用于利用所述硬伪标签和所述软伪标签对所述第一行人重识别网络模型和第二行人重识别网络模型进行协同训练;
筛选模块,用于选取两个时序平均模型中mAP精度值大的模型作为目标行人重识别网络模型。
6.如权利要求5所述的装置,其特征在于,所述提取特征模块包括:
处理单元,用于将所述目标域图像集输入所述每个行人重识别网络模型中,利用所述卷积神经网络进行处理,并进行空间维度展平得到特征f,将所述特征f附上cls标记得到特征fcls,并对所述特征fcls进行位置编码得到特征fcp;
转换单元,用于将所述特征fcp输入所述Transformer模块中,利用所述Transformer模块进行处理并进行转置和维度扩展,得到所述中间特征F。
7.一种行人重识别模型训练的设备,其特征在于,包括:
存储器,用于存储计算机程序;
处理器,用于执行所述计算机程序时实现如权利要求1至4任一项所述一种行人重识别模型训练方法的步骤。
8.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质上存储有计算机程序,所述计算机程序被处理器执行时实现如权利要求1至4任一项所述一种行人重识别模型训练方法的步骤。