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

Semantic segmentation work-flow #961

Merged
merged 4 commits into from
Aug 30, 2016

Conversation

gheinrich
Copy link
Contributor

No description provided.


### 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).
Copy link
Member

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://"?

Copy link
Contributor Author

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.

@gheinrich gheinrich force-pushed the dev/semantic-segmentation branch 2 times, most recently from 2f9afb9 to 4a4c71e Compare August 9, 2016 07:48
@lukeyeager
Copy link
Member

  • I can't view the labels DB for some reason
  • You should make a note that this requires NVcaffe 0.15 to avoid errors like Message type "caffe.LayerParameter" has no field named "crop_param".

@gheinrich
Copy link
Contributor Author

I can't view the labels DB for some reason

did you set the label encoding to 'png'?

you should make a note that this requires NVcaffe 0.15

Well spotted, thanks! (BVLC/master is also fine)

@lukeyeager
Copy link
Member

did you set the label encoding to 'png'?

Ah sorry, sloppy reading.

Got it to work! Cool stuff!

@gheinrich
Copy link
Contributor Author

I have added a note about the need for NVcaffe 0.15.

@jjmata
Copy link

jjmata commented Aug 28, 2016

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:

image

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?

*** SIGSEGV (@0xffffffff) received by PID 21336 (TID 0x7fe23d125a40) from PID 18446744073709551615; stack trace: ***
@     0x7fe23af54cb0 (unknown)
@     0x7fe21f30604a (unknown)
@     0x7fe21f314ae4 (unknown)
@     0x7fe21f276921 (unknown)
@     0x7fe21f26124d (unknown)
@     0x7fe23bfc05ce (unknown)
@     0x7fe23bfedc0c (unknown)
@     0x7fe23ca19715 caffe::GPUMemory::Manager::deallocate()
@     0x7fe23c9e3129 caffe::SyncedMemory::~SyncedMemory()
@     0x7fe23c998082 boost::detail::sp_counted_impl_p<>::dispose()
@     0x7fe23c98eda9 caffe::Blob<>::~Blob()
@     0x7fe23ca2c1b0 caffe::BasePrefetchingDataLayer<>::~BasePrefetchingDataLayer()
@     0x7fe23ca7abf9 caffe::DataLayer<>::~DataLayer()
@           0x40df96 std::vector<>::~vector()
@           0x40ec51 caffe::Net<>::~Net()
@           0x40ee19 caffe::Net<>::~Net()
@     0x7fe23c9ac310 caffe::SGDSolver<>::~SGDSolver()
@           0x40d05e boost::detail::sp_counted_base::release()
@           0x40aaf4 train()
@           0x4086bc main
@     0x7fe23af3ff45 (unknown)
@           0x408e8d (unknown)
@                0x0 (unknown)

@gheinrich
Copy link
Contributor Author

Hi @jjmata, this should work if you use dev/semantic-segmentation on my fork. Which version of Caffe are you using?

@lukeyeager
Copy link
Member

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."
Copy link
Member

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)"?

Copy link
Contributor Author

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.

@jjmata
Copy link

jjmata commented Aug 29, 2016

I'm on NVIDIA's 0.15.11 (and using your branch) ... sure 4GB should be enough? This is one of those dreadful AWS grid GPUs, FWIW!

screenshot 2016-08-29 15 46 32

@gheinrich
Copy link
Contributor Author

I'm on NVIDIA's 0.15.11 (and using your branch) ... sure 4GB should be enough?

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).

@gheinrich
Copy link
Contributor Author

@lukeyeager thanks for the review comments. I have updated the patch.

@lukeyeager
Copy link
Member

Thanks! Great stuff.

@lukeyeager lukeyeager merged commit f6b7df3 into NVIDIA:master Aug 30, 2016
@gheinrich gheinrich deleted the dev/semantic-segmentation branch November 30, 2016 16:50
SlipknotTN pushed a commit to cynnyx/DIGITS that referenced this pull request Mar 30, 2017
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants