公开(公告)号：CN118397611A

公开(公告)日：2024-07-26

申请号：CN202410567628.3

申请日：2024-05-09

Applicant: 中国地质大学(北京)

Inventor： 季晓慧 ,  吴宝坤 ,  杨眉 ,  何明跃 ,  张招崇 ,  曾姗

IPC: G06V20/60 ,  G06V10/26 ,  G06V10/44 ,  G06V10/52 ,  G06V10/70 ,  G06V10/80 ,  G06V10/82 ,  G06N3/0455 ,  G06N3/0464 ,  G06N3/084 ,  G06N3/0985

Abstract: 本发明实施例公开了一种基于图像分割的共生矿物识别方法与系统，所述方法与系统包括获取矿物图像并标注及扩充矿物图像分割数据集，构建以Swin Transformer为编码器、改进的特征金字塔为解码器的矿物图像分割模型，并使用多尺度金字塔池化及全局注意力模块以融合不同尺度的矿物特征信息，获取更全面和丰富的矿物特征表示，有效整合矿物图像的上下文信息，加强模型对于不同抽象级别特征的提取能力，保持特征图的丰富性，提高矿物分割性能，训练时使用迁移学习冻结编码器Swin Transformer在ImageNet预训练后在矿物分类数据集微调后的权重后对模型使用矿物分割数据集进行训练，所得模型解决了现有方法和系统仅能识别共生矿物图像中矿物种类，不能对各种类矿物进行准确定位的问题。