segNet: Add zero-copy class score getter #574
Open
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR adds a getter for class probabilities to segNet.
This getter is zero-copy on a Jetson since all it does is return a pointer to the memory area containing non-argmaxed raw output.
I noticed that segNet::Mask() and segNet::Overlay() do not argmax on GPU but on CPU, so there is no performance loss on performing argmax yourself and this can be useful if you want to do e.g. bilinear filtering on probabilities in order to get less "blocky" results. This is my use case, so I figured somebody else could use this.