Merge pull request clovaai#48 from YacobBY/patch-2

ku21fan · web-flow · commit d62daf3baa6a · 2019-08-03T15:04:38.000+09:00
replaced deprecated fromstring() with frombuffer()
diff --git a/create_lmdb_dataset.py b/create_lmdb_dataset.py
@@ -11,7 +11,7 @@
 def checkImageIsValid(imageBin):
     if imageBin is None:
         return False
-    imageBuf = np.fromstring(imageBin, dtype=np.uint8)
+    imageBuf = np.frombuffer(imageBin, dtype=np.uint8)
     img = cv2.imdecode(imageBuf, cv2.IMREAD_GRAYSCALE)
     imgH, imgW = img.shape[0], img.shape[1]
     if imgH * imgW == 0:
