Update create_dataset.py

bgshih · Dec 21, 2015 · 8bcb8bb · 8bcb8bb
1 parent bd6e82f
commit 8bcb8bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tool/create_dataset.py b/tool/create_dataset.py
@@ -6,8 +6,8 @@
 def checkImageIsValid(imageBin):
     if imageBin is None:
         return False
-    imageBuf = np.fromstring(imgBin, dtype=np.uint8)
-    img = cv2.imdecode(imgBuf, cv2.IMREAD_GRAYSCALE)
+    imageBuf = np.fromstring(imageBin, dtype=np.uint8)
+    img = cv2.imdecode(imageBuf, cv2.IMREAD_GRAYSCALE)
     imgH, imgW = img.shape[0], img.shape[1]
     if imgH * imgW == 0:
         return False