Fix Python wrapper for GpuMat

2018-09-28 08:58:30 +09:00
parent 8f6695acc7
commit 5f17cc1abd
1 changed files with 13 additions and 0 deletions
@@ -7,6 +7,19 @@ typedef cuda::GpuMat::Allocator GpuMat_Allocator;
 typedef cuda::HostMem::AllocType HostMem_AllocType;
 typedef cuda::Event::CreateFlags Event_CreateFlags;

+template<> struct pyopencvVecConverter<cuda::GpuMat>
+{
+    static bool to(PyObject* obj, std::vector<cuda::GpuMat>& value, const ArgInfo info)
+    {
+        return pyopencv_to_generic_vec(obj, value, info);
+    }
+
+    static PyObject* from(const std::vector<cuda::GpuMat>& value)
+    {
+        return pyopencv_from_generic_vec(value);
+    }
+};
+
 CV_PY_TO_CLASS(cuda::GpuMat);
 CV_PY_TO_CLASS(cuda::Stream);
 CV_PY_TO_CLASS(cuda::Event);