Image Segmentation Method of Rock Core Hole Based on MSHAM-UNet

Nan-yang WANG , Jiang-hai SHEN

Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (24) : 10362 -10369.

PDF
Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (24) : 10362-10369 DOI: 10.12404/j.issn.1671-1815.2306417
research-article
Image Segmentation Method of Rock Core Hole Based on MSHAM-UNet
Author information +
History +
PDF

Abstract

The hole segmentation of core images is important for oil exploration. In order to solve the problems of discontinuous hole edge segmentation, low segmentation accuracy and large number of parameters, an improved model based on UNet network MSHAM-UNet was proposed. Firstly, aiming at the problem of semantic information loss caused by the hopping connection of UNet models to feature maps of different scales, a multi-scale hybrid fusion attention module (MSHAM) combined with a hybrid attention mechanism (HAM) was designed, which performed attention feature fusion on shallow feature maps with spatial information and deep feature maps with semantic information, so as to enhanced the ability of the network to aggregate information at different scales. Secondly, the GP-bneck module was used to replace part of the ordinary convolution, which can reduce the number of model parameters and deepen the network, and enhance the network feature extraction ability. The experimental results show that the F1-score, intersection over union (IoU) and mean intersection over union (MIoU) of the MSHAM-UNet network on the core hole dataset reach 87.35%, 77.27% and 90.21%, respectively, which are 5.29%, 4.02% and 4.84% higher than the original model, and are also improved compared with the mainstream semantic segmentation model, which provides a new idea for core pore research.

Keywords

core hole images / deep learning / UNet model / attention mechanisms / feature fusion

Cite this article

Download citation ▾
Nan-yang WANG, Jiang-hai SHEN. Image Segmentation Method of Rock Core Hole Based on MSHAM-UNet. Science Technology and Engineering, 2024, 24(24): 10362-10369 DOI:10.12404/j.issn.1671-1815.2306417

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (11386KB)

18

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉