Merge pull request #16513 from pwuertz:cuda_py_interop

2020-03-05 11:56:07 +00:00
parent 23bc89d6fc 53c0189ed7
commit 90a4d67e8d
4 changed files with 29 additions and 0 deletions
@@ -26,5 +26,13 @@ class cuda_test(NewOpenCVTests):

        self.assertTrue(np.allclose(cuMat.download(), npMat))

+    def test_cuda_interop(self):
+        npMat = (np.random.random((128, 128, 3)) * 255).astype(np.uint8)
+        cuMat = cv.cuda_GpuMat()
+        cuMat.upload(npMat)
+        self.assertTrue(cuMat.cudaPtr() != 0)
+        stream = cv.cuda_Stream()
+        self.assertTrue(stream.cudaPtr() != 0)
+
 if __name__ == '__main__':
    NewOpenCVTests.bootstrap()