Rock Image Lithology Recognition Method Based on Lightweight Convolutional Neural Network
-
摘要: 【目的】岩性识别是油气勘探和开发过程中的重要环节,对于油气勘探定位、储层评价以及建立储层模型具有重要的指导意义。但传统的人工岩性识别方法耗时耗力、经典的深度学习模型虽然识别精度高,但模型的参数量较大,为了提高模型识别精度,同时降低模型的参数量,使模型适用于岩性实时识别工作。【方法】本文首先收集了白云岩、砂岩等8种共3016张岩石图像构建岩性识别数据集,然后以轻量型卷积神经网络ShuffleNetV2模型为基础网络,提出了一种Rock-ShuffleNetV2岩性识别模型(下文简称为RSHFNet模型)。模型中将混合注意力模块(Convolutional Block Attention Module,CBAM)以及多尺度特征融合模块(Multi Scale Feature Fusion Module,MSF)融入基础网络中以加强模型的特征提取能力,提升模型识别性能,并优化模型中ShuffleNetv2单元的堆叠次数以减少模型参数量。【结果】实验结果表明:与基础模型相比,本文提出的RSHFNet模型的准确率达到了87.21%,提高了4.98%;同时,模型参数量与浮点运算量降低到了8.69×10^6与9.3×10^7,分别是基础模型的67 %与63 %,模型参数量明显降低;并且RSHFNet模型的综合性能明显优于现有的卷积神经网络。【结论】本文提出的RSHFNet岩性识别模型具有较高的识别精度和较好的泛化能力,同时更加的轻量化,为实现野外实时的岩性识别工作提供了新思路。
-
关键词:
- 岩性识别 /
- ShuffleNetV2网络 /
- 混合注意力机制 /
- 多尺度特征融合模块
Abstract: 【Objective】Lithology identification is a crucial step in the process of oil and gas exploration and development, providing important guidance for exploration positioning, reservoir evaluation, and the establishment of reservoir models. However, traditional manual lithology identification methods are time-consuming and labor-intensive. Although classical deep learning models achieve high identification accuracy, they often have a large number of parameters. To enhance model accuracy while reducing the number of parameters, the aim is to make the model suitable for real-time lithology identification.【Methods】This paper first collected a dataset of 3,016 rock images consisting of eight types, including dolomite and sandstone. Based on the lightweight convolutional neural network ShuffleNetV2, the paper proposes a Rock-ShuffleNetV2 lithology identification model (hereafter referred to as the RSHFNet model). The model incorporates the Convolutional Block Attention Module (CBAM) and Multi-Scale Feature Fusion Module (MSF) into the base network to enhance feature extraction capabilities and improve identification performance. Additionally, the number of stacked ShuffleNetV2 units is optimized to reduce the model's parameters.【Results】The experimental results show that the RSHFNet model achieved an accuracy of 87.21%, which is a 4.98% improvement over the baseline model. Furthermore, the model's parameters and floating-point operations were reduced to 8.69 × 10^6 and 9.3 × 10^7, respectively, representing 67% and 63% of the baseline model. This reduction significantly decreases the model's size. Additionally, the RSHFNet model demonstrates superior overall performance compared to existing convolutional neural networks.【Conclusion】The proposed RSHFNet lithology identification model offers high recognition accuracy and strong generalization capabilities while being more lightweight, providing a new approach for real-time lithology identification in the field.
点击查看大图
计量
- 文章访问数: 44
- PDF下载量: 5
- 被引次数: 0