Computed-Tomography Image Segmentation of Cerebral Hemorrhage Based on Improved U-shaped Neural Network

In view of the problem of low segmentation accuracy caused by the multi-scale of the lesion location in Computed-Tomography (CT) images of cerebral hemorrhage, an image segmentation model based on Attention improved U-shaped neural Network plus (AU-Net+) is proposed. Firstly, the model uses the encoder in U-Net to encode the features of the CT image of cerebral hemorrhage, and applies the proposed Residual Octave Convolution (ROC) block to the jump connection part of the U-shaped neural network to make the features of different levels more blend well. Secondly, for the merged features, a hybrid attention mechanism is introduced to improve the feature extraction ability of the target area. Finally, the Dice loss function is improved to enhance further the feature learning of the model for small and medium-sized target regions in CT images of cerebral hemorrhage. To verify the performance of the model, the mIoU index is improved by 20.9%, 3.6%, 7.0%, 3.1% compared with U-Net, Attention U-Net, UNet++ and CE-Net respectively, which indicates that AU-Net+ model has better segmentation effect.