中药材识别模型的训练方法、中药材种类识别方法及装置

本申请公开了一种中药材识别模型的训练方法、中药材种类识别方法及装置，该训练方法包括获取训练数据集，所述训练数据集包括第一训练数据集和第二训练数据集；构建中药材识别模型，所述中药材识别模型包括主干网络和对比学习模块；将所述训练数据集输入到所述中药材识别模型，以通过所述主干网络得到所述第一训练数据集对应的检测损失，以及通过所述对比学习模块得到所述第二训练数据集对应的对比损失；交替采用所述检测损失和所述对比损失对所述中药材识别模型进行训练，直到满足预设收敛条件以得到训练好的中药材识别模型；由此，通过将自监督学习中的对比学习与识别模型相结合进行交替训练，以提高相似种类中药材的识别准确率。