基于多任务学习的卫星图像中太阳能光伏板视觉识别方法

本发明公开了一种基于多任务学习的卫星图像中太阳能光伏板视觉识别方法，属于机器视觉技术领域，所述方法包括：获取太阳能光伏板卫星图像数据集；其中，数据集中的卫星图像不仅包含语义分割标签，还包含太阳能光伏板的背景信息；构建多任务学习模型；其中，多任务学习模型的输入为太阳能光伏板卫星图像，输出为太阳能光伏板区域的分割结果以及太阳能光伏板的背景类别；利用数据集对多任务学习模型进行训练；利用训练好的多任务学习模型实现卫星图像中太阳能光伏板视觉识别。本发明方案可以自动从高分辨率卫星图像中提取特征，并准确识别出太阳能板的区域和背景信息，从而方便后续对地区发电能力的预测。