Skip to content

amartyacodes/Multiclass-U-Net-for-Fetal-Feature-Extraction

Repository files navigation

Multiclass-U-Net-for-Fetal-Feature-Extraction

Built a Multiclass U-Net Model for deep learning based segmentation mode capable of identifying the structures of the fetal transventricular plane Upon using the "ipynb" notebook the results can be following results can be achieved

Multiclass U-Net Model

Multiclass-input-and-multiclass-output-U-Net-schematic-Our-U-Net-has-4-channel-input\

The whole model has been implemented in the ipynb file from scratch with the help of TensorFlow 2.0 and Keras

For judging the performance Dice coefficient was used

The formula for Dice coefficient is = 2 * the Area of Overlap divided by the total number of pixels in both images. and is used mostly for segmentation problems

Dice Coefficient Graph Achieved

Dice Coeff Graph

Image - Labels - Prediction

Test Image Results

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published