recover

fwy423 · Apr 5, 2017 · 18191e7 · 18191e7
1 parent 955dc54
commit 18191e7
Show file tree

Hide file tree

Showing 6 changed files with 17 additions and 10 deletions.
diff --git a/my_params/checkpoint b/my_params/checkpoint
@@ -1,2 +1,2 @@
-model_checkpoint_path: "save_net.ckpt"
-all_model_checkpoint_paths: "save_net.ckpt"
+model_checkpoint_path: "light-model"
+all_model_checkpoint_paths: "light-model"
diff --git a/my_params/light-model.data-00000-of-00001 b/my_params/light-model.data-00000-of-00001
diff --git a/my_params/light-model.index b/my_params/light-model.index
diff --git a/my_params/light-model.meta b/my_params/light-model.meta
diff --git a/tensorflow_recover_CNN_lighter.py b/tensorflow_recover_CNN_lighter.py
@@ -218,7 +218,7 @@ def upsampling_layer(input_x, k_size):
 learning_rate = 1e-4
 
 batch_size = 128
-epoch = 50
+epoch = 80
 
 ####################
 # Building Network #
@@ -305,6 +305,12 @@ def upsampling_layer(input_x, k_size):
 
     sess.run(tf.global_variables_initializer())
 
+    # load the previous training result if it is possible
+    try:
+        saver.restore(sess, "my_params/light-model")
+    except Exception:
+        pass
+
     iter_total = 0
     for i in range(epoch):
         iters = int(train_input.shape[0] / batch_size)

diff --git a/tesser_ocr.py b/tesser_ocr.py
@@ -10,13 +10,14 @@ def batch_rec(img_path):
 
     dirs = os.listdir(img_path)
     for img in dirs:
-        # try:
-        #     content = tesserocr.file_to_text(img)
-        # except Exception:
-        #     print("skip", img)
-        #     continue
-
-        content = tesserocr.file_to_text(img_path+img)
+        print("predicting "+img)
+        try:
+            content = tesserocr.file_to_text(img)
+        except Exception:
+            print("skip", img)
+            continue
+
+        # content = tesserocr.file_to_text(img_path+img)
 
         name, _ = img.split(".")
         with open(img_path + name + '_rec.txt', 'w') as file: