Skip to content
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

The unpool is not standard right ? #9

Open
DrM175u opened this issue Mar 9, 2017 · 2 comments
Open

The unpool is not standard right ? #9

DrM175u opened this issue Mar 9, 2017 · 2 comments

Comments

@DrM175u
Copy link

DrM175u commented Mar 9, 2017

Hello,

from the code I have read in your project, it the unpool operation is just an image resize. So basically, this
network is not really a segnet network (at least not in a 'canonical' way from the paper).
Do you plan to use the locations of the max-pool layers to implement the standard unpool op ?

@jeroen-rooijmans
Copy link

@DrM175u I am currently working on this, there is also some discussion here: https://github.com/tensorflow/tensorflow/issues/2169. I will keep you up to date if I successfully implement the unpooling using the max indices. Note that you have to register the MaxPool gradient to MaxPoolWithArgmax because that is currently missing in Tensorflow.

@andreaazzini
Copy link
Owner

@DrM175u correct, the network I have implemented is not the canonical segnet. I have implemented the canonical version, but I decided to leave it outside the scope of the project. It was definitely painful to train it using the argmax-based unpooling layers and, as @jeroen-rooijmans pointed out, MaxPoolWithArgmax is not available for CPUs.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants