基于U-Net网络的遥感图像自动分割方法研究
本研究基于U-Net网络,尝试使用卷积神经网络自动分割遥感影像中的建筑。首先提出了一种基于遥感图像类别比率的交叉熵损失函数——类别平衡交叉熵,并与U-Net相结合,将其应用于遥感图像语义分割。在Inria Aerial Image Labeling Dataset训练数据集上分别使用交叉熵损失函数和类别平衡交叉熵损失函数进行训练,得到两个训练好的卷积神经网络。结果发现,交叉熵的F1 Score为0.47,类别平衡交叉熵的F1 Score为……
下载地址
用户评论