Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
U
Unet-Segmentation-Pytorch-Nest-of-Unets
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Wiki
Requirements
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Locked files
Build
Pipelines
Jobs
Pipeline schedules
Test cases
Artifacts
Deploy
Releases
Package registry
Container registry
Model registry
Operate
Environments
Terraform modules
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Code review analytics
Issue analytics
Insights
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
GitLab community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
manli
Unet-Segmentation-Pytorch-Nest-of-Unets
Commits
e8989cab
Unverified
Commit
e8989cab
authored
6 years ago
by
Malav Bateriwala
Committed by
GitHub
6 years ago
Browse files
Options
Downloads
Patches
Plain Diff
Update README.md
parent
abb5e1de
No related branches found
No related tags found
No related merge requests found
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
README.md
+15
-4
15 additions, 4 deletions
README.md
with
15 additions
and
4 deletions
README.md
+
15
−
4
View file @
e8989cab
...
@@ -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**
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment