-
Notifications
You must be signed in to change notification settings - Fork 1.4k
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
Semantic segmentation work-flow #961
Semantic segmentation work-flow #961
Conversation
|
||
### FCN-Alexnet | ||
|
||
In this example we will use [FCN-Alexnet](https://github.com/shelhamer/fcn.berkeleyvision.org/tree/master/voc-fcn-alexnet) from [fcn.berkeleyvision.org](fcn.berkeleyvision.org). |
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.
This link doesn't work. I guess it needs the "http://"?
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.
Thanks, updated in latest commit.
2f9afb9
to
4a4c71e
Compare
|
did you set the label encoding to 'png'?
Well spotted, thanks! (BVLC/master is also fine) |
Ah sorry, sloppy reading. Got it to work! Cool stuff! |
4a4c71e
to
c894015
Compare
I have added a note about the need for NVcaffe 0.15. |
c894015
to
87921ab
Compare
Not sure this is supposed to actually work yet (assume so?) but I've followed the steps and keep hitting this error after a few hours of training: Running from a fork of your repo, @gheinrich on the tip of the dev/semantic-segmentation branch. Sorry, should have probably checked this first ... the Caffe logs seem to indicate and out-of-memory situation in the GPU?
|
Hi @jjmata, this should work if you use |
Yeah, deallocating an invalid pointer sounds like a caffe bug. |
" image folder there must be one corresponding image in the label" | ||
" image folder. The label image must have the same filename except" | ||
" for the extension, which may differ. Label images are expected" | ||
" to be single-channel, palette images." |
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.
To match the above desc, should this be "single-channel images (paletted or grayscale)"?
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.
Sure, I will make the two tool tips the same.
87921ab
to
c88d568
Compare
I presume so. When training on VOC with the default batch size of 1, GPU memory utilization is under 2GB. I am using NV-Caffe 0.15.9 (the version that got released through .deb packages). |
@lukeyeager thanks for the review comments. I have updated the patch. |
Thanks! Great stuff. |
Semantic segmentation work-flow
No description provided.