Skip to content

MICCAI Challenge 2023: Pubic Symphysis-Fetal Head Segmentation task solved using the U-Net model from the SMP library, accompanied by an algorithm for Angle of Progression estimation.

Notifications You must be signed in to change notification settings

an-eve/FH-PS-AOP-2023-segmentation-challenge

Repository files navigation

MICCAI 2023 FH-PS-AOP Challenge

Goal of the project

Addressing the Pubic Symphysis-Fetal Head Segmentation and Angle of Progression Challenge, the repository focuses on automating the segmentation of the transperineal ultrasound images.

The ultimate objective is to enhance the accuracy and objectivity of fetal head descent assessment, crucial for optimizing obstetric practices and minimizing difficult vaginal deliveries. The challenge encourages the development and application of cutting-edge techniques for FH-PS segmentation, serving as a benchmark dataset for comprehensive evaluation and fostering advancements in the field.

The dataset

The JNU-IFM dataset for segmenting pubic symphysis-fetal head is originally introduced in the article.

This segmentation competition exclusively utilizes MHA (MetaImage) files for image data, comprising a dataset of 4000 samples. The original images are in a $3\times256\times256$ format, while the ground truth labels have a shape format of $256\times256$.

The label pixels are categorized as follows:

  • 0: Background

  • 1: Pubic Symphysis

  • 2: Fetal Head

Dataset sample

Architecture

  • A U-Net model (PyTorch SMP library) with mit_b0 encoder (transformer) and imagenet pre-trained weights for encoder initialization.
  • The file named segmentation_model.pth contains the weight parameters of the trained model.
  • Execute the FH_PS_AOP_Challenge.ipynb notebook to train the designated model.

Results

  • Sensitivity score: 0.986
  • Specificity score: 0.993
  • Pixel accuracy: 0.991
  • Jaccard score: 0.973
  • Dice score: 0.986

Segmentation result

Angle of Progression Estimation Algoritm

The implementation details are in the file aop_estimation.py.

Angle

Next updates

Implement from scratch a ViT model, experiment with existing ones

About

MICCAI Challenge 2023: Pubic Symphysis-Fetal Head Segmentation task solved using the U-Net model from the SMP library, accompanied by an algorithm for Angle of Progression estimation.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published