-
Notifications
You must be signed in to change notification settings - Fork 1.1k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Feature] Support FLAVR Net #866
Conversation
|
||
# Batch mean normalization works slightly better than global mean | ||
# normalization, Refer to https://github.com/myungsub/CAIN | ||
mean_ = images.mean( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
image.mean(dim=(2,3,4), True)
Codecov Report
@@ Coverage Diff @@
## master #866 +/- ##
==========================================
+ Coverage 83.12% 83.25% +0.13%
==========================================
Files 220 222 +2
Lines 12453 12648 +195
Branches 2017 2049 +32
==========================================
+ Hits 10351 10530 +179
- Misses 1785 1794 +9
- Partials 317 324 +7
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
dx_out = self.decoder(xs) | ||
|
||
out = self.feature_fuse(dx_out) | ||
out = self.conv_last(out) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I meant we'd better add some comments here to indicate out's shape.
|
||
# prepare data | ||
inputs = torch.rand(1, 4, 3, 256, 248) | ||
target = torch.rand(1, 3, 256, 248) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
248 or 256?
|
||
UpConv3d(4, 4, 1, 1, 1) | ||
conv3d = UpConv3d(4, 4, 1, 1, 1, up_mode='trilinear', batchnorm=True) | ||
conv3d(inputs) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Any assertion for line 42-46?
* [Feature] Support FLAVR Net * Update * Update * Update * Update * Update
* [Feature] Support FLAVR Net * Update * Update * Update * Update * Update
* [Feature] Support FLAVR Net * Update * Update * Update * Update * Update
No description provided.