工业图像处理模型的预训练方法、装置及电子设备

本申请公开了一种工业图像处理模型的预训练方法、装置及电子设备，属于图像处理技术领域。方法包括：对无标注的工业图像样本集进行掩码处理，得到第一样本集和第二样本集，获得第一训练分支输出的第一工业视觉特征，获得第二训练分支输出的第二工业视觉特征；基于第一工业视觉特征和第二工业视觉特征，对第一训练分支和第二训练分支进行对比学习，获得第一训练分支和第二训练分支之间的目标损失函数，更新第一训练分支和第二训练分支的模型参数，生成工业图像处理模型的预训练模型参数。该方法可以充分捕获工业数据多样化的视觉特征，有效提升模型的泛化能力和表达能力。