未验证 提交 a2fc2385 编写于 作者: Q qingqing01 提交者: GitHub

Merge pull request #459 from alexqdh/develop

fix recognize_digits infer
...@@ -321,7 +321,7 @@ def load_image(file): ...@@ -321,7 +321,7 @@ def load_image(file):
im = Image.open(file).convert('L') im = Image.open(file).convert('L')
im = im.resize((28, 28), Image.ANTIALIAS) im = im.resize((28, 28), Image.ANTIALIAS)
im = np.array(im).astype(np.float32).flatten() im = np.array(im).astype(np.float32).flatten()
im = im / 255.0 im = im / 255.0 * 2.0 - 1.0
return im return im
test_data = [] test_data = []
......
...@@ -325,7 +325,7 @@ def load_image(file): ...@@ -325,7 +325,7 @@ def load_image(file):
im = Image.open(file).convert('L') im = Image.open(file).convert('L')
im = im.resize((28, 28), Image.ANTIALIAS) im = im.resize((28, 28), Image.ANTIALIAS)
im = np.array(im).astype(np.float32).flatten() im = np.array(im).astype(np.float32).flatten()
im = im / 255.0 im = im / 255.0 * 2.0 - 1.0
return im return im
test_data = [] test_data = []
......
...@@ -363,7 +363,7 @@ def load_image(file): ...@@ -363,7 +363,7 @@ def load_image(file):
im = Image.open(file).convert('L') im = Image.open(file).convert('L')
im = im.resize((28, 28), Image.ANTIALIAS) im = im.resize((28, 28), Image.ANTIALIAS)
im = np.array(im).astype(np.float32).flatten() im = np.array(im).astype(np.float32).flatten()
im = im / 255.0 im = im / 255.0 * 2.0 - 1.0
return im return im
test_data = [] test_data = []
......
...@@ -367,7 +367,7 @@ def load_image(file): ...@@ -367,7 +367,7 @@ def load_image(file):
im = Image.open(file).convert('L') im = Image.open(file).convert('L')
im = im.resize((28, 28), Image.ANTIALIAS) im = im.resize((28, 28), Image.ANTIALIAS)
im = np.array(im).astype(np.float32).flatten() im = np.array(im).astype(np.float32).flatten()
im = im / 255.0 im = im / 255.0 * 2.0 - 1.0
return im return im
test_data = [] test_data = []
......
...@@ -112,7 +112,7 @@ def main(): ...@@ -112,7 +112,7 @@ def main():
im = Image.open(file).convert('L') im = Image.open(file).convert('L')
im = im.resize((28, 28), Image.ANTIALIAS) im = im.resize((28, 28), Image.ANTIALIAS)
im = np.array(im).astype(np.float32).flatten() im = np.array(im).astype(np.float32).flatten()
im = im / 255.0 im = im / 255.0 * 2.0 - 1.0
return im return im
test_data = [] test_data = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册