基于压缩模型的遥感图像场景分类方法和装置

    公开(公告)号:CN118015350A

    公开(公告)日:2024-05-10

    申请号:CN202410111293.4

    申请日:2024-01-26

    Abstract: 本发明公开了一种基于压缩模型的遥感图像场景分类方法和装置,所述方法包括:获取遥感图像训练数据集;所述遥感图像训练数据集,包括遥感图像和对应的分类标签信息;利用所述遥感图像训练数据集,对预设的遥感图像分类网络进行训练处理,得到训练后的遥感图像分类网络;对所述训练后的遥感图像分类网络进行压缩处理,得到压缩遥感图像分类网络;利用所述压缩遥感图像分类网络对所采集的遥感图像进行场景分类处理,得到所述遥感图像的场景类型信息。本发明对模型的滤波器进行压缩,压缩后的模型仍然保持原有的结构形式,在推理计算时,无需特定的加速库的支持;压缩后的模型的内存占用将大大减少,而且推理运算速度大幅度提升,对搭载平台的硬件性能要求更低。

Patent Agency Ranking