Skip to content
Snippets Groups Projects
Unverified Commit e8989cab authored by Malav Bateriwala's avatar Malav Bateriwala Committed by GitHub
Browse files

Update README.md

parent abb5e1de
No related branches found
No related tags found
No related merge requests found
...@@ -45,8 +45,19 @@ Install all dependent libraries: ...@@ -45,8 +45,19 @@ Install all dependent libraries:
```bash ```bash
pip install -r requirements.txt pip install -r requirements.txt
``` ```
## 3. Run the file
## 3. Types of Unet Add all your folders to this location
```
t_data = '' # Input data
l_data = '' #Input Label
test_image = '' #Image to be predicted while training
test_label = '' #Label of the prediction Image
test_folderP = '' #Test folder Image
test_folderL = '' #Test folder Label for calculating the Dice score
```
## 4. Types of Unet
**Unet** **Unet**
![unet1](/images/unet1.png) ![unet1](/images/unet1.png)
...@@ -67,7 +78,7 @@ Install all dependent libraries: ...@@ -67,7 +78,7 @@ Install all dependent libraries:
![nested](/images/nested.jpg) ![nested](/images/nested.jpg)
## 4. Visualization ## 5. Visualization
To plot the loss , Visdom would be required. The code is already written, just uncomment the required part. To plot the loss , Visdom would be required. The code is already written, just uncomment the required part.
Gradient flow can be used too. Taken from (https://discuss.pytorch.org/t/check-gradient-flow-in-network/15063/10) Gradient flow can be used too. Taken from (https://discuss.pytorch.org/t/check-gradient-flow-in-network/15063/10)
...@@ -97,7 +108,7 @@ Still have to tweak some parameters to get visualization. Have messed up this tr ...@@ -97,7 +108,7 @@ Still have to tweak some parameters to get visualization. Have messed up this tr
<img src="https://github.com/bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets/blob/master/images/in2.png" width="480"> <img src="https://github.com/bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets/blob/master/images/in2.png" width="480">
## 5. Results ## 6. Results
**Dice Score for hippocampus segmentation** **Dice Score for hippocampus segmentation**
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment