jetson-utils icon indicating copy to clipboard operation
jetson-utils copied to clipboard

Published 20 hours ago • verified publisher icon dusty-nv

Native VPI Support

Open ligaz opened this issue 2 years ago • 1 comments

I'm experimenting with the VPI API and was wondering do you plan to support it "natively" without the need to go through the jetson.utils.cudaToNumpy() hoops?

It looks like the vpi.asimage(...) method is expecting an object with __cuda_array_interface__ so if PyCudaImage has one it will work out of the box.

ligaz avatar May 20 '22 07:05 ligaz

Hi @ligaz, I haven't looked into this yet but think it's a good idea. I think a lot of it will depend on if I'm able to implement the numba __cuda_array_interface__ from C or not. Maybe it will be enough to just implement the dict as they wish...

dusty-nv avatar May 20 '22 12:05 dusty-nv

@dusty-nv It looks like this has been implemented in https://github.com/dusty-nv/jetson-inference/commit/8b499817381cd7ac3f72c7c8742e6a29c1b696c0

ligaz avatar Dec 05 '22 09:12 ligaz

Thanks @ligaz, yes it has been implemented, along with the PyCUDA gpudata interface

dusty-nv avatar Dec 05 '22 14:12 dusty-nv