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.
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:
- Create a Kaggle Account: If you don't have a Kaggle account, you will need to create one.
- Accept the Competition Rules: Before downloading the dataset, you will need to accept the competition rules.
- Download the Dataset: Once you have accepted the rules, you can download the dataset by clicking the "Download All" button on the data page.
-
Clone the repository
git clone https://github.com/rakeshutekar/Image-Segmentation-with-U-Netification.git
-
Install Dependencies
pip install -r requirements.txt
Training the Model Navigate to the src directory and run the training script:
- Train
cd src python train.py - Evaluating the Model After training, you can evaluate the model: python evaluate.py
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
