来源:期刊VIP网 时间:
作者:程国建;李宗祥;李秋实;韩江;孙亚招;
单位:西安石油大学计算机学院;中国石油长庆油田公司;长庆油田分公司第八采油厂;
摘要:为了实现岩石薄片岩性高效、准确的分类与识别,提出了一种基于空间注意力与多尺度融合的岩石薄片岩性识别方法。采用多尺度融合的策略,通过多个EfficientNet中的轻量反转瓶颈卷积核(MBConv)对橄榄石、普通辉石、角闪石、黑云母等多种类别的岩石薄片图像进行特征提取,以捕获更多的细节信息。引入空间注意力模块(SGE),融合岩石薄片图像中的空间特征信息。此外,采用Ranger优化器,改善模型的性能及收敛速度。实验表明:提出的MFSRE(Multi-Scale Fusion-SGE-Ranger-EfficientNet)模型在测试集上的召回率、F1分数分别为98.25%、98.29%,具有较高的识别准确率,相较于ShuffleNet、RegNet、MobileNetV2网络具有更好的分类效果。
关键词:岩性识别;;岩石薄片图像;;EfficientNet;;空间注意力机制
基金资助:国家自然科学基金项目(40872087);; 陕西省自然科学基金(2023-JC-YB-601);; 西安市科技计划高校院所人才服务企业项目(23GXFW0077);; 陕西省“两链”融合重点专项—两链融合秦创原总窗口产业集群项目(2022QCY-LL-82)