No commit message

pannous · Jan 10, 2019 · e5ec6ac · e5ec6ac
1 parent b62317c
commit e5ec6ac
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/test_image.png b/test_image.png
diff --git a/text_recognizer.py b/text_recognizer.py
@@ -7,13 +7,12 @@
 from keras.models import load_model
 from PIL import Image # python -m pip install --upgrade Pillow  # WTF
 
+weight_file = None # use model weights
 # weight_file = 'best_weights.h5'
 # weight_file = 'current_weights.h5'
-
-weight_file = 'weights_ascii.h5' # learned on noisy data
-# weight_file = 'weights_ascii_easy.h5' # no freckles
-# weight_file = 'weights_ascii_clean.h5' # pure text
-# weight_file = None # use model weights
+# weight_file = 'weights_ascii.h5' # learned on noisy data
+weight_file = 'weights_ascii_easy.h5' # no freckles
+# weight_file = 'weights_ascii_clean.h5' # pure text, no rotation etc
 
 chars = u'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZäöüÄÖÜß0123456789!@#$%^&*()[]{}-_=+\\|"\'`;:/.,?><~ '
 
@@ -25,7 +24,8 @@ def init_model(model_file="current_model.h5"):
   global model
   model = load_model(model_file)
   model.summary()
-  model.load_weights(weight_file, reshape=True, by_name=True)
+  if weight_file:
+    model.load_weights(weight_file, reshape=True, by_name=True)
 
 def predict_tensor(tensor):
   if len(tensor.shape) == 2: