火源定位方法、装置、模型训练方法、计算机设备及介质

本发明涉及计算机视觉技术领域，公开了一种火源定位方法、装置、模型训练方法、计算机设备及介质。火源检测定位模型包括图像特征提取模块、注意力机制模块和半监督学习模块。首先，获取待检测场景的目标图像。接下来，将目标图像输入至火源检测定位模型中，通过图像特征提取模块提取目标图像的图像特征，通过注意力机制模块基于图像特征自动聚焦于图像中的关键区域，特别是可能的火源位置，从而实现对火源的快速和准确定位，得到火源区域特征。通过半监督学习模块利用未标注数据和标注数据，更新火源检测定位模型的参数，有效减少对数据标注的依赖，缓解数据标注费用高昂的问题，同时提升了模型的泛化能力，得到火源定位信息。