Enable imshow to take GpuMat inputs in Python

alejoeie · Oct 19, 2020 · bf49149 · bf49149
1 parent c8ebe0e
commit bf49149
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/modules/python/src2/hdr_parser.py b/modules/python/src2/hdr_parser.py
@@ -958,7 +958,9 @@ def parse(self, hname, wmode=True):
                         else:
                             decls.append(decl)
 
-                            if self._generate_gpumat_decls and "cv.cuda" in decl[0]:
+                            if self._generate_gpumat_decls and ("cv.cuda" in decl[0] or decl[0] in [
+                                "cv.imshow", # https://github.com/opencv/opencv/issues/18553
+                            ]):
                                 # If function takes as one of arguments Mat or vector<Mat> - we want to create the
                                 # same declaration working with GpuMat
                                 args = decl[3]