Skip to content

rakeshutekar/Image-Segmentation-with-U-Net

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Image Segmentation with U-Net

This project demonstrates image segmentation using the U-Net architecture for medical image analysis. The dataset used is the Data Science Bowl 2018 dataset, which consists of images of cell nuclei.

Dataset

The dataset used in this project is the Data Science Bowl 2018: Nuclei Detection dataset from Kaggle. It consists of various images of cell nuclei and their corresponding masks. Each image has one or more masks indicating the location of nuclei within the image.

To use the dataset:

  1. Create a Kaggle Account: If you don't have a Kaggle account, you will need to create one.
  2. Accept the Competition Rules: Before downloading the dataset, you will need to accept the competition rules.
  3. Download the Dataset: Once you have accepted the rules, you can download the dataset by clicking the "Download All" button on the data page.

Setup

  1. Clone the repository

    git clone https://github.com/rakeshutekar/Image-Segmentation-with-U-Netification.git
  2. Install Dependencies

    pip install -r requirements.txt
    

Running the Project

Training the Model Navigate to the src directory and run the training script:

  1. Train
    cd src
    python train.py
    
    
  2. Evaluating the Model After training, you can evaluate the model: python evaluate.py

Results:

  5/5 ━━━━━━━━━━━━━━━━━━━━ 2s 449ms/step - accuracy: 0.9478 - loss: 0.0930
  Test Loss: 0.09798252582550049
  Test Accuracy: 0.9455899000167847
  5/5 ━━━━━━━━━━━━━━━━━━━━ 3s 514ms/step
  Mean IoU: 0.7401531363366687

output

About

This project demonstrates image segmentation using the U-Net architecture for medical image analysis. The dataset used is the Data Science Bowl 2018 dataset, which consists of images of cell nuclei.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages