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

摘要

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