--- a/README.md +++ b/README.md @@ -1,46 +1,46 @@ -# 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! +# 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!