Journal of Optoelectronics · Laser, Volume. 34, Issue 12, 1337(2023)

Prostate magnetic resonance image segmentation based on improved 3D UNet

SANG Zijiang1, SHAO Yeqin2、*, and XU Changyan1
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less

    An improved 3D UNet network model that combines a dual path attention (DPA) module and a multi-scale feature aggregation (MFA) module is proposed to address the problems such as blurred edges,low contrast,and uneven gray value distribution in magnetic resonance (MR) images of the prostate bringing about the poor segmentation accuracy.Firstly,the dataset is resampled and cropped to fit the model input.Then,DPA and residual connection are added to each layer of the 3D UNet network encoder to enhance the feature coding capability.At the same time,an MFA module is added to the network decoder to make full use of spatial context information and enhance semantic information.Finally,the proposed model is validated on the public dataset PROMISE12,with the Dice coefficient of 89.90% and the Hausdorff distance of 9.37 mm. Compared with other models,the proposed model has better segmentation results,and the number of parameters and the amount of computation are less.

    Tools

    Get Citation

    Copy Citation Text

    SANG Zijiang, SHAO Yeqin, XU Changyan. Prostate magnetic resonance image segmentation based on improved 3D UNet[J]. Journal of Optoelectronics · Laser, 2023, 34(12): 1337

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Received: Dec. 27, 2022

    Accepted: --

    Published Online: Sep. 25, 2024

    The Author Email: SHAO Yeqin (hnsyk@ntu.edu.cn)

    DOI:10.16136/j.joel.2023.12.0861

    Topics