基于轻量级多尺度Transformer网络的高光谱图像分类方法及系统

    公开(公告)号:CN117710818A

    公开(公告)日:2024-03-15

    申请号:CN202311760700.6

    申请日:2023-12-20

    Abstract: 本发明公开了基于轻量级多尺度Transformer网络的高光谱图像分类方法及系统,包括:对输入的高光谱图像进行数据预处理,提取带有标签的图像块;划分训练集和测试集,构建高光谱图像分类模型,包括:提取图像块的初步特征,通过不同尺度的轻量级Transformer编码器提取特征,利用全局平均池化操作将特征转换为一维向量,进而利用全连接层进行分类;训练高光谱图像分类模型,进而根据高光谱图像分类模型,精确识别高光谱图像中的地物目标。本发明的基于轻量级多尺度Transformer网络的高光谱图像分类方法,能够降低模型的参数量,在缓解过拟合现象的同时取得相似甚至更好的性能,以实现更加快速、高精度的地物识别。

Patent Agency Ranking