# UNET Segmentation on CT Scan Images

Dataset - [Kaggle Dataset Link](https://www.kaggle.com/datasets/nikhilroxtomar/ct-heart-segmentation)

### How to Train the Model

```

1. python3 -m pip install --user virtualenv (In Mac or Linux)

   python -m pip install --user virtualenv (In Windows) 

   [Make sure Python3 is installed in your system]

2. python3 -m venv env (In Mac or Linux)

   python -m venv env (In Windows) 

3. source env/bin/activate  (In Mac or Linux)

   .\env\Scripts\activate (In Windows) 

4. pip3 install --upgrade pip (In Mac or Linux)

   pip install --upgrade pip (In Windows)

4. pip3 install -r requirements.txt (In Mac or Linux)

   pip install -r requirements.txt (In Windows)

```

> Formatter Used - black

### Project Structure

```

.

├── README.md

├── dataloader

│   └── data.py

├── eval.py

├── logs

├── metrics.py

├── models

│   └── unet.py

├── predict.py

├── requirements.txt

├── results

├── saved_models

└── train.py

```

Thank You!