potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 15:23:05 +04:00
Code Packages Releases Wiki Activity

cuda: add python bindings to allow GpuMat and Stream objects to be initialized from raw pointers

Browse Source
This commit is contained in:
cudawarped
2023-03-17 16:31:51 +02:00
parent 97c021b17a
commit 7539abecdb
3 changed files with 45 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/python/test/test_cuda.py
+6
View File
@@ -40,8 +40,14 @@ class cuda_test(NewOpenCVTests):
cuMat = cv.cuda_GpuMat()
cuMat.upload(npMat)
self.assertTrue(cuMat.cudaPtr() != 0)
cuMatFromPtrSz = cv.cuda.createGpuMatFromCudaMemory(cuMat.size(),cuMat.type(),cuMat.cudaPtr(), cuMat.step)
self.assertTrue(cuMat.cudaPtr() == cuMatFromPtrSz.cudaPtr())
cuMatFromPtrRc = cv.cuda.createGpuMatFromCudaMemory(cuMat.size()[1],cuMat.size()[0],cuMat.type(),cuMat.cudaPtr(), cuMat.step)
self.assertTrue(cuMat.cudaPtr() == cuMatFromPtrRc.cudaPtr())
stream = cv.cuda_Stream()
self.assertTrue(stream.cudaPtr() != 0)
streamFromPtr = cv.cuda.wrapStream(stream.cudaPtr())
self.assertTrue(stream.cudaPtr() == streamFromPtr.cudaPtr())
asyncstream = cv.cuda_Stream(1) # cudaStreamNonBlocking
self.assertTrue(asyncstream.cudaPtr() != 0)