基于TransUNet的岩石薄片图像粒度识别方法、电子设备及存储介质

    公开(公告)号:CN116543256A

    公开(公告)日:2023-08-04

    申请号:CN202310439646.9

    申请日:2023-04-23

    Abstract: 本公开涉及一种基于TransUNet的岩石薄片图像粒度识别方法、电子设备及存储介质,所述方法包括:采集岩心薄片图像、数据预处理、特征提取、特征映射、全局上下文建模、逐像素分类、计算损失、迭代训练以及重复执行直到损失函数停止降低时停止训练,得到岩石薄片图像粒度识别transUnet训练模型;利用得到的岩石薄片图像粒度识别transUnet训练模型对DataB数据集中图像进行粒度识别,得到粒度识别后的岩石薄片颗粒表征图像;所述全局上下文建模包括在TransUNet解码器中,进行全局上下文建模,生成增强特征图S3,使用基于Transformer的结构对经由特征映射后获得的特征图S2进行处理,以捕捉特征图S2中岩石薄片图像的全局长距离依赖关系,所述处理包括通过自注意力机制、多头自注意力、层归一化、残差连接以及前馈神经网络对特征图S2中的岩石薄片图像进行处理和分割;本公开实施例能够实现精确的特征提取和全局上下文建模,提高岩石薄片粒度识别的准确性。

Patent Agency Ranking