The invention discloses a
coal rock fracture intelligent extraction method based on improved U-Net. The method comprises the following steps: S1, constructing a
coal rock fracture CT image
data set; s2, constructing an improved U-Net segmentation model, specifically comprising the following steps: S2.1, taking VGG16 as a
backbone network, and introducing a depth separable
convolution module; s2.2, a PPA attention module is added after each layer of depth separable
convolution of the decoder, the PPA attention module is introduced after each up-sampling stage of the decoder, and the output of the PPA attention module is subjected to batch normalization and Dropout layer
processing; s2.3, defining a composite
loss function; s3, training and optimizing a segmentation model, wherein the specific steps comprise: S3.1, setting hyper-parameters; and S3.2, training the model by using the
training set, adjusting hyper-parameters by using the
verification set, and evaluating the performance by using the
test set, wherein the evaluation indexes comprise MIoU, MAcc and FWIoU. According to the method, the problems of difficult identification of small fractures,
large model calculation amount, poor multi-scale
information fusion and
class imbalance in the
coal rock fracture image can be solved, and the robustness, segmentation precision and practicability of the model are improved.