期刊
COMPUTERS & GEOSCIENCES
卷 154, 期 -, 页码 -出版社
PERGAMON-ELSEVIER SCIENCE LTD
DOI: 10.1016/j.cageo.2021.104799
关键词
Deep learning; Rock images; Intelligent detection; Lithology identification
资金
- National Science Fund for Excellent Young Scholars [52022053]
- Science Fund for Distinguished Young Scholars of Shandong Province [ZR201910270116]
- China Postdoctoral Science Foundation [2019M662361]
An intelligent lithology identification method based on deep learning using a rock detection model built on Faster R-CNN and ResNet structures has been proposed, achieving higher accuracy and stability than traditional models in terms of speed, accuracy, and identification capability.
An intelligent lithology identification method is proposed based on the deep learning of rock images. The lithology information and position information in rock images can be predicted using the Faster R-CNN architecture through the RPN proposal generation algorithm and the Fast R-CNN detector. To obtain more rock features, the rock detection model is built on the ResNet structure, and the residual learning is used to retain as much as possible detailed information in the original input image. The four-step alternating training is used to fine-tuned end-to-end, and the prediction results are optimized by the cross-entropy loss and the regression loss. To speed up the model and improve the identification accuracy, data augmentation and pre-training are used to train the model. The mAP, P, R and F-1 score are used as evaluation indexes of the accuracy, and the Faster R-CNN model is compared with the YOLO v4 model. Results indicate that the mAP of the rock detection model based on the Faster R-CNN is 99.19% and the F-1 score is 96.6%. Compared with the YOLO v4 model, the accuracy is higher and the identification ability is more stable. The proposed rock detection model has good identification ability for different rocks in rock images, and the model is of good robustness and generalization performance, which is suitable for rapid intelligent lithology identification in practical geological and logging engineering.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据