Update classifier.py

TroddenSpade · Jul 9, 2021 · b1cc9ed · b1cc9ed
1 parent e5c8e08
commit b1cc9ed
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/classifier.py b/src/classifier.py
@@ -3,7 +3,7 @@
 import numpy as np
 import cv2
 
-model = tf.keras.models.load_model('models/spc1.h5')
+model = tf.keras.models.load_model('models/spc3.h5')
 
 def classify(X):
     res = model.predict(X)
@@ -12,4 +12,7 @@ def classify(X):
 def preprocess(img):
     img = img.astype(np.float32) / 255
     resized_image = cv2.resize(img, (10, 28))
+    # means = resized_image.mean(axis=(0,1), dtype='float64')
+    # std = resized_image.std(axis=(0,1), dtype='float64')
+    # normalized_img = (resized_image - means) / std
     return resized_image