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.

Sign up for GitHub

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

Jump to bottom

Add Python binding to another overload of segNet::Mask #476

Open

hanzhi713 wants to merge 2 commits into dusty-nv:master from MARS-UVA:master

hanzhi713 commented Jan 11, 2020 •

edited

Addresses issue #473

Add binding to another overload of segNet::Mask
Use integer division in segnet-camera.py

Usage of the new python binding:

# classes is an uint8 array
classes = jetson.utils.cudaAllocMapped(width * height)
net.MaskClass(classes, width, height)
# cudaToNumpy assumes float32. Need to dividie output length by 4.
mask = jetson.utils.cudaToNumpy(classes, 1, width * height // 4, 1)[:, 0, 0]
# convert resulting float32 view to uint8 view
mask_np = mask.view('uint8').reshape(width, height)

hanzhi713 added 2 commits

January 11, 2020 14:51


          integral parameters

1b9e501


          binding for another overload of segNet::Mask

cd16f62

hanzhi713 changed the title ~~Add binding to another overload of segNet::Mask~~ Add Python binding to another overload of segNet::Mask

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment