一种基于联邦学习的山火图像识别方法及相关装置

本发明实施例公开了一种基于联邦学习的山火图像识别方法及相关装置，其中，该方法包括：通过将山火图像训练样本划分K类不同山火类型下的训练样本，得到K类训练样本，使得同一类型的山火图像作为一组训练样本，对山火图像分类联邦学习框架下的本地训练模型进行全局训练，获取待识别的山火特征图像，将待识别的山火特征图像输入全局收敛后的本地训练模型，输出识别结果。在本实施例中，可以基于不同类型山火的训练样本集中对山火图像联邦学习框架进行集中训练，避免了异构数据造成模型识别精度下降，并使得训练后山火图像联邦学习框架可以识别各个类型的山火图像，以提高山火图像联邦学习框架的识别精度。