-
Notifications
You must be signed in to change notification settings - Fork 45.4k
DenseNet and FC-DenseNet #2308
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
DenseNet and FC-DenseNet #2308
Conversation
Only implemented very simple tests for now. Tests run OK.
|
/cc @ikhlestov |
|
why closed? |
|
Can one of the admins verify this patch? |
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.
Somehow I was mentioned in the review comments. I've checked the images DenseNet and it seems to me ok. However, It will be great if someone can use network with various input shapes, not depending on the last average pooling with 7x7 filters.
| # We perform a first convolution. | ||
| # If DenseNet BC, first convolution has 2*growth_rate output channels | ||
| if bottleneck and compression < 1.0: | ||
| n_filters_first_conv = 2 * growth_rate |
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 think this behavior should be somehow be mentioned in the densenet method documentation.
| ##################### | ||
| pool_name = 'pool%d' % (n_dense + 1) | ||
| if dense_prediction: | ||
| net = slim.pool(net, [7, 7], pooling_type='AVG', scope=pool_name) |
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.
What if we have not 224x224 input? From my point of view we should use dynamic average pooling based on input shape.
| from __future__ import division | ||
| from __future__ import print_function | ||
|
|
||
| import numpy as np |
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.
Just unused import
I implemented slim models for DenseNet and FC-DenseNet, based on the original lasagne/theano code made available by the authors of FC-DenseNets. Hope it will be useful.
DenseNets for Semantic Segmentation - code