|
a/README.md |
|
b/README.md |
1 |
# UNET Segmentation on CT Scan Images |
1 |
# UNET Segmentation on CT Scan Images |
2 |
|
2 |
|
3 |
Dataset - [Kaggle Dataset Link](https://www.kaggle.com/datasets/nikhilroxtomar/ct-heart-segmentation) |
3 |
Dataset - [Kaggle Dataset Link](https://www.kaggle.com/datasets/nikhilroxtomar/ct-heart-segmentation) |
4 |
|
4 |
|
5 |
### How to Train the Model |
5 |
### How to Train the Model |
6 |
|
6 |
|
7 |
``` |
7 |
```
|
8 |
1. python3 -m pip install --user virtualenv (In Mac or Linux) |
8 |
1. python3 -m pip install --user virtualenv (In Mac or Linux)
|
9 |
python -m pip install --user virtualenv (In Windows) |
9 |
python -m pip install --user virtualenv (In Windows)
|
10 |
|
10 |
|
11 |
[Make sure Python3 is installed in your system] |
11 |
[Make sure Python3 is installed in your system] |
12 |
|
12 |
|
13 |
2. python3 -m venv env (In Mac or Linux) |
13 |
2. python3 -m venv env (In Mac or Linux)
|
14 |
python -m venv env (In Windows) |
14 |
python -m venv env (In Windows)
|
15 |
|
15 |
|
16 |
3. source env/bin/activate (In Mac or Linux) |
16 |
3. source env/bin/activate (In Mac or Linux)
|
17 |
.\env\Scripts\activate (In Windows) |
17 |
.\env\Scripts\activate (In Windows) |
18 |
|
18 |
|
19 |
4. pip3 install --upgrade pip (In Mac or Linux) |
19 |
4. pip3 install --upgrade pip (In Mac or Linux)
|
20 |
pip install --upgrade pip (In Windows) |
20 |
pip install --upgrade pip (In Windows)
|
21 |
|
21 |
|
22 |
4. pip3 install -r requirements.txt (In Mac or Linux) |
22 |
4. pip3 install -r requirements.txt (In Mac or Linux)
|
23 |
pip install -r requirements.txt (In Windows) |
23 |
pip install -r requirements.txt (In Windows)
|
24 |
``` |
24 |
``` |
25 |
|
25 |
|
26 |
> Formatter Used - black |
26 |
Formatter Used - black |
27 |
|
27 |
|
28 |
### Project Structure |
28 |
### Project Structure
|
29 |
``` |
29 |
```
|
30 |
. |
30 |
.
|
31 |
├── README.md |
31 |
├── README.md
|
32 |
├── dataloader |
32 |
├── dataloader
|
33 |
│ └── data.py |
33 |
│ └── data.py
|
34 |
├── eval.py |
34 |
├── eval.py
|
35 |
├── logs |
35 |
├── logs
|
36 |
├── metrics.py |
36 |
├── metrics.py
|
37 |
├── models |
37 |
├── models
|
38 |
│ └── unet.py |
38 |
│ └── unet.py
|
39 |
├── predict.py |
39 |
├── predict.py
|
40 |
├── requirements.txt |
40 |
├── requirements.txt
|
41 |
├── results |
41 |
├── results
|
42 |
├── saved_models |
42 |
├── saved_models
|
43 |
└── train.py |
43 |
└── train.py
|
44 |
``` |
44 |
``` |
45 |
|
45 |
|
46 |
Thank You! |
46 |
Thank You!
|