The repository contains the code for UNET segmentation on CT scan dataset in TensorFlow 2.0 framework.
YouTube video: UNET Segmentation on CT Scan Images using TensorFlow 2.0
- Dataset
- Architecture
- Results
- Contact
The dataset contains the CT scan image and their respective binary mask. These binary mask are annotated using VGG Image Annotator website.
Download the dataset: CT Heart Dataset - 2D Binary Segmentation Task
Original Image | Mask Image |
---|---|
The block diagram of the UNET architecture taken from the original paper.
U-Net Architecture |
The images below contains:
- Input image
- Ground truth
- Predicted mask
For any query contact below: