基于边缘约束单图生成模型的不平衡数据增强方法

本发明公开的基于边缘约束单图生成模型的不平衡数据增强方法，属于人工智能生成技术与遥感交叉技术领域。本发明实现方法为：1：对目标区域进行光学探测，获取图像数据集；2、利用单图去噪扩散模型对少数类目标数据X进行训练生成多样化目标数据；3、利用全局边缘约束方法对多样化目标数据修正，调整生成目标的多样性和遵循原始目标轮廓特征之间的平衡性；4、结合单图去噪扩散模型与全局边缘约束方法的目标函数，进行联合学习以实现数据增强，获得大量少数类目标数据；5、基于数据增强后的数据集训练深度学习识别模型，输出对待识别数据目标类别的识别结果。与现有技术相比，本发明提升目标识别中少数类目标的识别精度。