mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-12-18 11:28:02 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Wrap gpuMat::release to Python

Browse Source
This commit is contained in:
huangziqing 2022-05-02 00:54:17 +08:00
parent 1b07274e34
commit 82ae9ef541
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/core/include/opencv2/core/cuda.hpp
View File

@ -155,7 +155,7 @@ public:
CV_WRAP void create(Size size, int type); CV_WRAP void create(Size size, int type);
//! decreases reference counter, deallocate the data when reference counter reaches 0 //! decreases reference counter, deallocate the data when reference counter reaches 0
void release(); CV_WRAP void release();
//! swaps with other smart pointer //! swaps with other smart pointer
CV_WRAP void swap(GpuMat& mat); CV_WRAP void swap(GpuMat& mat);

9
modules/python/test/test_cuda.py
View File

@ -55,5 +55,14 @@ class cuda_test(NewOpenCVTests):
self.assertEqual(cuMat.size(), (1024, 1024)) self.assertEqual(cuMat.size(), (1024, 1024))
self.assertEqual(cuMat.type(), cv.CV_8UC3) self.assertEqual(cuMat.type(), cv.CV_8UC3)
def test_cuda_release(self):
npMat = (np.random.random((128, 128, 3)) * 255).astype(np.uint8)
cuMat = cv.cuda_GpuMat()
cuMat.upload(npMat)
cuMat.release()
self.assertTrue(cuMat.cudaPtr() == 0)
self.assertTrue(cuMat.step == 0)
self.assertTrue(cuMat.size() == (0, 0))
if __name__ == '__main__': if __name__ == '__main__':
NewOpenCVTests.bootstrap() NewOpenCVTests.bootstrap()