Skip to content

Commit d8fd279

Browse files
committed
Fix #5 PyMat.get_data type conversion error
1 parent e596d80 commit d8fd279

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pyzed/core.pyx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -388,10 +388,10 @@ cdef class PyMat:
388388

389389
cdef size_t size = 0
390390
dtype = None
391-
if self.mat.getDataType() == MAT_TYPE_8U_C1 or MAT_TYPE_8U_C2 or MAT_TYPE_8U_C3 or MAT_TYPE_8U_C4:
391+
if self.mat.getDataType() in (MAT_TYPE_8U_C1, MAT_TYPE_8U_C2, MAT_TYPE_8U_C3, MAT_TYPE_8U_C4):
392392
size = self.mat.getHeight()*self.mat.getWidth()*self.mat.getChannels()
393393
dtype = np.uint8
394-
elif self.mat.getDataType() == MAT_TYPE_32F_C1 or MAT_TYPE_32F_C2 or MAT_TYPE_32F_C3 or MAT_TYPE_32F_C4:
394+
elif self.mat.getDataType() in (MAT_TYPE_32F_C1, MAT_TYPE_32F_C2, MAT_TYPE_32F_C3, MAT_TYPE_32F_C4):
395395
size = self.mat.getHeight()*self.mat.getWidth()*self.mat.getChannels()*sizeof(float)
396396
dtype = np.float32
397397
else:

0 commit comments

Comments
 (0)