Abstract:The basic identification and classification of sedimentary rocks into sandstone and mudstone are important in the study of sedimentology and they are executed by a sedimentologist. However, such manual activity involves countless hours of observation and data collection prior to any interpretation. When such activity is conducted in the field as part of an outcrop study, the sedimentologist is likely to be exposed to challenging conditions such as the weather and their accessibility to the outcrops. This study uses high-resolution photographs which are acquired from a sedimentological study to test an alternative basic multi-rock identification through machine learning. While existing studies have effectively applied deep learning techniques to classify the rock types in field rock images, their approaches only handle a single rock-type classification per image. One study applied deep learning techniques to classify multi-rock types in each image; however, the test was performed on artificially overlaid images of different rock types in a test sample and not of naturally occurring rock surfaces of multiple rock types. To the best of our knowledge, no study has applied semantic segmentation to solve the multi-rock classification problem using digital photographs of multiple rock types. This paper presents the application of two state-of-the-art segmentation models, namely U-Net and LinkNet, to identify multiple rock types in digital photographs by segmenting the sandstone, mudstone, and background classes in a self-collected dataset of 102 images from a field in Brunei Darussalam. Four pre-trained networks, including Resnet34, Inceptionv3, VGG16, and Efficientnetb7 were used as a backbone for both models, and the performances of the individual models and their ensembles were compared. We also investigated the impact of image enhancement and different color representations on the performances of these segmentation models. The experiment results of this study show that among the individual models, LinkNet with Efficientnetb7 as a backbone had the best performance with a mean over intersection (MIoU) value of 0.8135 for all of the classes. While the ensemble of U-Net models (with all four backbones) performed slightly better than the LinkNet with Efficientnetb7 did with an MIoU of 0.8201. When different color representations and image enhancements were explored, the best performance (MIoU = 0.8178) was noticed for the L*a*b* color representation with Efficientnetb7 using U-Net segmentation. For the individual classes of interest (sandstone and mudstone), U-Net with Efficientnetb7 was found to be the best model for the segmentation. Thus, this study presents the potential of semantic segmentation in automating the reservoir characterization process whereby we can extract the patches of interest from the rocks for much deeper study and modeling to be conducted.

Influence of Neural Network Structure on Rock Intelligent Classification Based on Structural and Tectonic Features of Rocks

Self-intersection Attention Pooling Based Classification for Rock Recognition

Deep learning based classification of rock structure of tunnel face

Deep learning of rock microscopic images for intelligent lithology identification: Neural network comparison and selection

Rock image classification using deep residual neural network with transfer learning

Rock Classification from Field Image Patches Analyzed Using a Deep Convolutional Neural Network

Research on Rock Image Recognition Based on Deep Learning

Autonomous and Real-time Rock Image Classification using Convolutional Neural Networks

AI-based rock strength assessment from tunnel face images using hybrid neural networks

Deep learning of rock images for intelligent lithology identification

Rock Classification Based on Residual Networks

A Novel Method of Multitype Hybrid Rock Lithology Classification Based on Convolutional Neural Networks

Design of Rock Sample Classification and Recognition Technology Based on Deep Learning

Rock thin sections identification based on improved squeeze-and-Excitation Networks model

Automatic lithology identification method based on efficient deep convolutional network

A Deep Learning Model for Quick and Accurate Rock Recognition with Smartphones

Autonomous Martian rock image classification based on transfer deep learning methods

Experimental Studies on Rock Thin-Section Image Classification by Deep Learning-Based Approaches

Segmentation for Multi-Rock Types on Digital Outcrop Photographs Using Deep Learning Techniques

Enhancing Rock Image Segmentation in Digital Rock Physics: A Fusion of Generative AI and State-of-the-Art Neural Networks