Skip to content

The repository contains the code for UNET segmentation on CT scan dataset in TensorFlow 2.0 framework.

Notifications You must be signed in to change notification settings

nikhilroxtomar/UNET-Segmentation-on-CT-Scan-Images-in-TensorFlow-2.0

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

UNET Segmentation on CT Scan Images in TensorFlow 2.0

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

Overview

  • Dataset
  • Architecture
  • Results
  • Contact

Dataset

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

Architecture

The block diagram of the UNET architecture taken from the original paper.

U-Net Architecture
U-Net Architecture

Results

The images below contains:

  1. Input image
  2. Ground truth
  3. Predicted mask

Contact

For any query contact below:

About

The repository contains the code for UNET segmentation on CT scan dataset in TensorFlow 2.0 framework.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages