-
When trying to use As a workaround, I am sending data to CPU, augmenting it and then sending back to GPU. This is horribly slow. |
Beta Was this translation helpful? Give feedback.
Replies: 2 comments 2 replies
-
@EmilPi it would require integrating the opencv GPU support in mxnet image API to use it natively with GPU arrays. For now you can consider using the OpenCV python API with GPU support for data augmentation, and then use from_dlpack and to_dlpack_for_write for zero-copy loading and sharing of ndarray. |
Beta Was this translation helpful? Give feedback.
-
Is there any example of sharing mxnet <-> OpenCV GpuMat data? What am I doing wrong? I am trying to
But this code
Fails with I also tried
but this fails with
|
Beta Was this translation helpful? Give feedback.
@EmilPi it would require integrating the opencv GPU support in mxnet image API to use it natively with GPU arrays. For now you can consider using the OpenCV python API with GPU support for data augmentation, and then use from_dlpack and to_dlpack_for_write for zero-copy loading and sharing of ndarray.