change the input to float

microsoft · May 24, 2018 · d4861dd · d4861dd
1 parent 15e821b
commit d4861dd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/mmdnn/conversion/examples/imagenet_test.py b/mmdnn/conversion/examples/imagenet_test.py
@@ -229,7 +229,7 @@ def __init__(self):
     def ZeroCenter(path, size, BGRTranspose=False):
         img = Image.open(path)
         img = img.resize((size, size))
-        x = np.array(img, dtype=np.float64)
+        x = np.array(img, dtype=np.float32)
 
         # Reference: 1) Keras image preprocess: https://github.com/keras-team/keras/blob/master/keras/applications/imagenet_utils.py
         #            2) tensorflow github issue: https://github.com/tensorflow/models/issues/517
@@ -250,7 +250,7 @@ def ZeroCenter(path, size, BGRTranspose=False):
     def Normalize(path, size=224, scale=0.0392156863 ,mean=[-0.485, -0.456, -0.406], std=[0.229, 0.224, 0.225], BGRTranspose = False):
         img = Image.open(path)
         img = img.resize((size, size))
-        x = np.array(img, dtype=np.float64)
+        x = np.array(img, dtype=np.float32)
         x *= scale
         for i in range(0, 3):
             x[..., i] += mean[i]
@@ -264,7 +264,7 @@ def Normalize(path, size=224, scale=0.0392156863 ,mean=[-0.485, -0.456, -0.406],
     def Standard(path, size, BGRTranspose=False):
         img = Image.open(path)
         img = img.resize((size, size))
-        x = np.array(img, dtype=np.float64)
+        x = np.array(img, dtype=np.float32)
         x /= 255.0
         x -= 0.5
         x *= 2.0
@@ -277,7 +277,7 @@ def Standard(path, size, BGRTranspose=False):
     def Identity(path, size, BGRTranspose=False):
         img = Image.open(path)
         img = img.resize((size, size))
-        x = np.array(img, dtype=np.float64)
+        x = np.array(img, dtype=np.float32)
         if BGRTranspose == True:
             x = x[..., ::-1]
         return x

diff --git a/mmdnn/visualization/index.html b/mmdnn/visualization/index.html
@@ -48,7 +48,7 @@ <h4 >Tips</h4>
                 <div>
                     <input type="text" id="uname" name="name"
                             placeholder="Search for Layer Name"
-                            size="30"  required>
+                            size="20"  required>
                 </div>
 
             </li>