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

replaced deprecated fromstring() with frombuffer()
WJinhai · Aug 3, 2019 · d62daf3 · d62daf3
2 parents 9128831 + cbc1d9d
commit d62daf3
Showing 1 changed file with 1 addition and 1 deletion.
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: