From c8df372f633a8ce3976e27b6a714cfd143196c0b Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Thu, 6 Jun 2019 23:47:37 +0800
Subject: [PATCH] network runs

---
 .gitignore                                    |   1 +
 .idea/workspace.xml                           | 227 +++++++++++-------
 dataset/__pycache__/casia_hwdb.cpython-36.pyc | Bin 0 -> 2718 bytes
 dataset/casia_hwdb.py                         |  16 +-
 dataset/dataset_hwdb.py                       |   0
 models/__pycache__/cnn_net.cpython-36.pyc     | Bin 0 -> 2865 bytes
 models/cnn_net.py                             |  38 ++-
 train.py                                      |  48 ++--
 8 files changed, 212 insertions(+), 118 deletions(-)
 create mode 100755 dataset/__pycache__/casia_hwdb.cpython-36.pyc
 delete mode 100755 dataset/dataset_hwdb.py
 create mode 100755 models/__pycache__/cnn_net.cpython-36.pyc

diff --git a/.gitignore b/.gitignore
index 1d74e21..ea1ac33 100755
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,2 @@
 .vscode/
+checkpoints/
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index bdd3ca6..1b0bbe6 100755
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -1,20 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
   <component name="ChangeListManager">
-    <list default="true" id="c86d3061-c2c8-42bb-882f-53f6373c7f88" name="Default" comment="">
-      <change beforePath="$PROJECT_DIR$/dataset/.gitignore" afterPath="$PROJECT_DIR$/dataset/.gitignore" />
-      <change beforePath="$PROJECT_DIR$/dataset/casia_hwdb.py" afterPath="$PROJECT_DIR$/dataset/casia_hwdb.py" />
-      <change beforePath="$PROJECT_DIR$/dataset/casia_hwdb_1.0_1.1.tfrecord" afterPath="$PROJECT_DIR$/dataset/casia_hwdb_1.0_1.1.tfrecord" />
-      <change beforePath="$PROJECT_DIR$/dataset/charactors.txt" afterPath="$PROJECT_DIR$/dataset/charactors.txt" />
-      <change beforePath="$PROJECT_DIR$/dataset/convert_to_tfrecord.py" afterPath="$PROJECT_DIR$/dataset/convert_to_tfrecord.py" />
-      <change beforePath="$PROJECT_DIR$/dataset/get_hwdb_1.0_1.1.sh" afterPath="$PROJECT_DIR$/dataset/get_hwdb_1.0_1.1.sh" />
-      <change beforePath="$PROJECT_DIR$/readme.md" afterPath="$PROJECT_DIR$/readme.md" />
-      <change beforePath="$PROJECT_DIR$/sample.png" afterPath="$PROJECT_DIR$/sample.png" />
-      <change beforePath="$PROJECT_DIR$/samples/.gitignore" afterPath="$PROJECT_DIR$/samples/.gitignore" />
-      <change beforePath="$PROJECT_DIR$/samples/001-f.gnt" afterPath="$PROJECT_DIR$/samples/001-f.gnt" />
-      <change beforePath="$PROJECT_DIR$/samples/sample.png" afterPath="$PROJECT_DIR$/samples/sample.png" />
-      <change beforePath="$PROJECT_DIR$/tests.py" afterPath="$PROJECT_DIR$/tests.py" />
-    </list>
+    <list default="true" id="c86d3061-c2c8-42bb-882f-53f6373c7f88" name="Default" comment="" />
     <option name="EXCLUDED_CONVERTED_TO_IGNORED" value="true" />
     <option name="TRACKING_ENABLED" value="true" />
     <option name="SHOW_DIALOG" value="false" />
@@ -23,29 +10,51 @@
     <option name="LAST_RESOLUTION" value="IGNORE" />
   </component>
   <component name="FileEditorManager">
-    <leaf>
-      <file leaf-file-name="dataset_hwdb.py" pinned="false" current-in-tab="false">
-        <entry file="file://$PROJECT_DIR$/dataset/dataset_hwdb.py">
+    <leaf SIDE_TABS_SIZE_LIMIT_KEY="300">
+      <file leaf-file-name="casia_hwdb.py" pinned="false" current-in-tab="false">
+        <entry file="file://$PROJECT_DIR$/../../dataset/casia_hwdb.py">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="0">
-              <caret line="0" column="0" lean-forward="false" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
+            <state relative-caret-position="418">
+              <caret line="54" column="35" lean-forward="false" selection-start-line="54" selection-start-column="35" selection-end-line="54" selection-end-column="35" />
               <folding />
             </state>
           </provider>
         </entry>
       </file>
-      <file leaf-file-name="casia_hwdb.py" pinned="false" current-in-tab="true">
-        <entry file="file://$PROJECT_DIR$/dataset/casia_hwdb.py">
+      <file leaf-file-name="train.py" pinned="false" current-in-tab="true">
+        <entry file="file://$PROJECT_DIR$/../../train.py">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="715">
-              <caret line="70" column="26" lean-forward="true" selection-start-line="70" selection-start-column="26" selection-end-line="70" selection-end-column="26" />
+            <state relative-caret-position="401">
+              <caret line="86" column="51" lean-forward="true" selection-start-line="86" selection-start-column="51" selection-end-line="86" selection-end-column="51" />
               <folding />
             </state>
           </provider>
         </entry>
       </file>
+      <file leaf-file-name=".gitignore" pinned="false" current-in-tab="false">
+        <entry file="file://$PROJECT_DIR$/../../.gitignore">
+          <provider selected="true" editor-type-id="text-editor">
+            <state relative-caret-position="17">
+              <caret line="1" column="12" lean-forward="false" selection-start-line="1" selection-start-column="12" selection-end-line="1" selection-end-column="12" />
+              <folding />
+            </state>
+          </provider>
+        </entry>
+      </file>
+      <file leaf-file-name="cnn_net.py" pinned="false" current-in-tab="false">
+        <entry file="file://$PROJECT_DIR$/../../models/cnn_net.py">
+          <provider selected="true" editor-type-id="text-editor">
+            <state relative-caret-position="646">
+              <caret line="38" column="0" lean-forward="false" selection-start-line="38" selection-start-column="0" selection-end-line="38" selection-end-column="0" />
+              <folding>
+                <element signature="e#1153#1176#0" expanded="true" />
+              </folding>
+            </state>
+          </provider>
+        </entry>
+      </file>
       <file leaf-file-name="tests.py" pinned="false" current-in-tab="false">
-        <entry file="file://$PROJECT_DIR$/tests.py">
+        <entry file="file://$PROJECT_DIR$/../../tests.py">
           <provider selected="true" editor-type-id="text-editor">
             <state relative-caret-position="0">
               <caret line="0" column="0" lean-forward="false" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
@@ -55,9 +64,9 @@
         </entry>
       </file>
       <file leaf-file-name="convert_to_tfrecord.py" pinned="false" current-in-tab="false">
-        <entry file="file://$PROJECT_DIR$/dataset/convert_to_tfrecord.py">
+        <entry file="file://$PROJECT_DIR$/../../dataset/convert_to_tfrecord.py">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="-301">
+            <state relative-caret-position="0">
               <caret line="0" column="0" lean-forward="false" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
               <folding />
             </state>
@@ -67,12 +76,16 @@
     </leaf>
   </component>
   <component name="Git.Settings">
-    <option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$" />
+    <option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$/../.." />
   </component>
   <component name="IdeDocumentHistory">
     <option name="CHANGED_PATHS">
       <list>
         <option value="$PROJECT_DIR$/dataset/casia_hwdb.py" />
+        <option value="$PROJECT_DIR$/../../.gitignore" />
+        <option value="$PROJECT_DIR$/../../models/cnn_net.py" />
+        <option value="$PROJECT_DIR$/../../dataset/casia_hwdb.py" />
+        <option value="$PROJECT_DIR$/../../train.py" />
       </list>
     </option>
   </component>
@@ -82,11 +95,10 @@
     <detection-done>true</detection-done>
     <sorting>DEFINITION_ORDER</sorting>
   </component>
-  <component name="ProjectFrameBounds">
-    <option name="x" value="933" />
-    <option name="y" value="28" />
-    <option name="width" value="1538" />
-    <option name="height" value="1412" />
+  <component name="ProjectFrameBounds" extendedState="4">
+    <option name="y" value="25" />
+    <option name="width" value="1573" />
+    <option name="height" value="1415" />
   </component>
   <component name="ProjectView">
     <navigator currentView="ProjectPane" proportions="" version="1">
@@ -103,24 +115,19 @@
       <foldersAlwaysOnTop value="true" />
     </navigator>
     <panes>
-      <pane id="Scratches" />
+      <pane id="Scope" />
       <pane id="ProjectPane">
         <subPane>
           <expand>
             <path>
-              <item name="ocrcn_tf2" type="b2602c69:ProjectViewProjectNode" />
+              <item name="." type="b2602c69:ProjectViewProjectNode" />
               <item name="ocrcn_tf2" type="462c0819:PsiDirectoryNode" />
             </path>
-            <path>
-              <item name="ocrcn_tf2" type="b2602c69:ProjectViewProjectNode" />
-              <item name="ocrcn_tf2" type="462c0819:PsiDirectoryNode" />
-              <item name="dataset" type="462c0819:PsiDirectoryNode" />
-            </path>
           </expand>
           <select />
         </subPane>
       </pane>
-      <pane id="Scope" />
+      <pane id="Scratches" />
     </panes>
   </component>
   <component name="PropertiesComponent">
@@ -157,22 +164,23 @@
     <servers />
   </component>
   <component name="ToolWindowManager">
-    <frame x="933" y="28" width="1538" height="1412" extended-state="0" />
+    <frame x="0" y="25" width="1573" height="1415" extended-state="4" />
     <editor active="true" />
     <layout>
-      <window_info id="TODO" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
-      <window_info id="Event Log" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="true" content_ui="tabs" />
-      <window_info id="Version Control" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
-      <window_info id="Python Console" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
-      <window_info id="Run" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
-      <window_info id="Terminal" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
-      <window_info id="Project" active="false" anchor="left" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="true" show_stripe_button="true" weight="0.22386059" sideWeight="0.5" order="-1" side_tool="false" content_ui="combo" />
-      <window_info id="Docker" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="false" weight="0.33" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
-      <window_info id="Database" active="false" anchor="right" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
-      <window_info id="SciView" active="false" anchor="right" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
-      <window_info id="Structure" active="false" anchor="left" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="true" content_ui="tabs" />
-      <window_info id="Favorites" active="false" anchor="left" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="true" content_ui="tabs" />
-      <window_info id="Debug" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
+      <window_info id="TODO" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="0" side_tool="false" content_ui="tabs" />
+      <window_info id="Event Log" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="0" side_tool="true" content_ui="tabs" />
+      <window_info id="Run" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="0" side_tool="false" content_ui="tabs" />
+      <window_info id="Version Control" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="0" side_tool="false" content_ui="tabs" />
+      <window_info id="Python Console" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="0" side_tool="false" content_ui="tabs" />
+      <window_info id="Terminal" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="true" show_stripe_button="true" weight="0.32943925" sideWeight="0.5" order="0" side_tool="false" content_ui="tabs" />
+      <window_info id="Project" active="false" anchor="left" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="true" show_stripe_button="true" weight="0.14800262" sideWeight="0.5" order="0" side_tool="false" content_ui="combo" />
+      <window_info id="Docker" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="false" weight="0.33" sideWeight="0.5" order="0" side_tool="false" content_ui="tabs" />
+      <window_info id="Database" active="false" anchor="right" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="0" side_tool="false" content_ui="tabs" />
+      <window_info id="Find" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.32943925" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
+      <window_info id="SciView" active="false" anchor="right" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="0" side_tool="false" content_ui="tabs" />
+      <window_info id="Structure" active="false" anchor="left" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="0" side_tool="true" content_ui="tabs" />
+      <window_info id="Debug" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="0" side_tool="false" content_ui="tabs" />
+      <window_info id="Favorites" active="false" anchor="left" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="0" side_tool="true" content_ui="tabs" />
     </layout>
   </component>
   <component name="TypeScriptGeneratedFilesManager">
@@ -186,6 +194,38 @@
     <watches-manager />
   </component>
   <component name="editorHistoryManager">
+    <entry file="file://$PROJECT_DIR$/../../dataset/dataset_hwdb.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="0">
+          <caret line="0" column="0" lean-forward="false" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
+          <folding />
+        </state>
+      </provider>
+    </entry>
+    <entry file="file://$PROJECT_DIR$/../../tests.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="0">
+          <caret line="0" column="0" lean-forward="false" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
+          <folding />
+        </state>
+      </provider>
+    </entry>
+    <entry file="file://$PROJECT_DIR$/../../dataset/convert_to_tfrecord.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="0">
+          <caret line="0" column="0" lean-forward="false" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
+          <folding />
+        </state>
+      </provider>
+    </entry>
+    <entry file="file://$PROJECT_DIR$/../../dataset/casia_hwdb.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="1190">
+          <caret line="70" column="26" lean-forward="true" selection-start-line="70" selection-start-column="26" selection-end-line="70" selection-end-column="26" />
+          <folding />
+        </state>
+      </provider>
+    </entry>
     <entry file="file:///usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/ops.py">
       <provider selected="true" editor-type-id="text-editor">
         <state relative-caret-position="-7666">
@@ -194,44 +234,19 @@
         </state>
       </provider>
     </entry>
-    <entry file="file://$PROJECT_DIR$/dataset/charactors.txt">
+    <entry file="file://$PROJECT_DIR$/../../dataset/charactors.txt">
       <provider selected="true" editor-type-id="text-editor">
         <state relative-caret-position="0">
           <caret line="0" column="0" lean-forward="false" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
-          <folding />
         </state>
       </provider>
     </entry>
-    <entry file="file://$PROJECT_DIR$/dataset/convert_to_tfrecord.py">
-      <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="-301">
-          <caret line="0" column="0" lean-forward="false" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
-          <folding />
-        </state>
-      </provider>
-    </entry>
-    <entry file="file://$PROJECT_DIR$/dataset/dataset_hwdb.py">
-      <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="0">
-          <caret line="0" column="0" lean-forward="false" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
-          <folding />
-        </state>
-      </provider>
-    </entry>
-    <entry file="file://$PROJECT_DIR$/train.py">
-      <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="0">
-          <caret line="0" column="0" lean-forward="false" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
-          <folding />
-        </state>
-      </provider>
-    </entry>
-    <entry file="file://$PROJECT_DIR$/sample.png">
+    <entry file="file://$PROJECT_DIR$/../../sample.png">
       <provider selected="true" editor-type-id="images">
         <state />
       </provider>
     </entry>
-    <entry file="file://$PROJECT_DIR$/tests.py">
+    <entry file="file://$PROJECT_DIR$/../../tests.py">
       <provider selected="true" editor-type-id="text-editor">
         <state relative-caret-position="0">
           <caret line="0" column="0" lean-forward="false" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
@@ -239,10 +254,52 @@
         </state>
       </provider>
     </entry>
-    <entry file="file://$PROJECT_DIR$/dataset/casia_hwdb.py">
+    <entry file="file://$PROJECT_DIR$/../../.gitignore">
       <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="715">
-          <caret line="70" column="26" lean-forward="true" selection-start-line="70" selection-start-column="26" selection-end-line="70" selection-end-column="26" />
+        <state relative-caret-position="17">
+          <caret line="1" column="12" lean-forward="false" selection-start-line="1" selection-start-column="12" selection-end-line="1" selection-end-column="12" />
+          <folding />
+        </state>
+      </provider>
+    </entry>
+    <entry file="file://$PROJECT_DIR$/../../dataset/dataset_hwdb.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="0">
+          <caret line="0" column="0" lean-forward="false" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
+          <folding />
+        </state>
+      </provider>
+    </entry>
+    <entry file="file://$PROJECT_DIR$/../../dataset/convert_to_tfrecord.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="0">
+          <caret line="0" column="0" lean-forward="false" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
+          <folding />
+        </state>
+      </provider>
+    </entry>
+    <entry file="file://$PROJECT_DIR$/../../dataset/casia_hwdb.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="418">
+          <caret line="54" column="35" lean-forward="false" selection-start-line="54" selection-start-column="35" selection-end-line="54" selection-end-column="35" />
+          <folding />
+        </state>
+      </provider>
+    </entry>
+    <entry file="file://$PROJECT_DIR$/../../models/cnn_net.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="646">
+          <caret line="38" column="0" lean-forward="false" selection-start-line="38" selection-start-column="0" selection-end-line="38" selection-end-column="0" />
+          <folding>
+            <element signature="e#1153#1176#0" expanded="true" />
+          </folding>
+        </state>
+      </provider>
+    </entry>
+    <entry file="file://$PROJECT_DIR$/../../train.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="401">
+          <caret line="86" column="51" lean-forward="true" selection-start-line="86" selection-start-column="51" selection-end-line="86" selection-end-column="51" />
           <folding />
         </state>
       </provider>
diff --git a/dataset/__pycache__/casia_hwdb.cpython-36.pyc b/dataset/__pycache__/casia_hwdb.cpython-36.pyc
new file mode 100755
index 0000000000000000000000000000000000000000..6a8613f0e5d0e3a16b7832d68b81bc9d94700306
GIT binary patch
literal 2718
zcmZ8i&5s+m6(>2Ln$>>TUOSSVqyr==s>0PK>!3&hXJIE>yG?>c5*u|fNHE|IDQRSB
zMkcB3t{|P{YY_BW<Xp7>M34C+cnZ+dUULYLocbQMyS8^EKE8Q;L%xsSd%Uy0p8Wg$
zU%r|pg#3@Z^aQ})gQ0&0B1lFA6>LPuEMt^Nwj>lK-pI>5H~U%O<}eH09A&Ya8(Bj(
zhRJZv(WYEq(k0C{9unb;;5iY2++2FumIy`koMcx-EE+Ih6^U4b`I_7o&8JNLB(E+>
zc3oZrdgJ&|ULU?DV5fcZ_2aKt*2>zlHS}SvC4=Ej*%Dhzkjh&ieSJy6-nLt9%Nw%Y
zCZK&www*j2z9BYcYl%LE!#4%kM#-~(J|V(;L^@Yy-zA9|lsbh!PmdIzOr%N&yb>ds
zetiGYU+?FCd9wdeD!AcVnnagXA=6rw{j%aCC&o{UdOR7G1vgS8M>3s)T0eFB79*~8
zYU;EvP1>D+81_y_{7{b4o~p;G;zvohUm3SolAJ;^NQc^JJT)>my@LV;=MMws9t@oV
zktCy#Hzp`#hzSp(>91lAnF_(X$Tpn|=)uE-$FmQU6rTI3JFZuPhhu&MwpWIim4=t<
zBrV6hFCi9SWHI217p7L)eOKV&_~4A=V;CAxMdox)a)1yVNe{qwua{3c%=%i6dddfd
z{}2_z^y^1EV<}3$Gb}4p@@l7&h8H`$+$pQ7{)U^fu6F7|6%{zXyK@%vor3F<=Yu10
z*qxl%IM2(fG<kjtk9RiF+4YzG?4D~B6jbnOz-Yq2Sn3~v2)dw#%*hZSzhvM2uwcjk
zHEih3A&he>{BOO5H}{6VzWq6wdm?Bf4ckOSd*n6bW8fQSd;<I$@(UkUnof89Tz3(G
zY{SVmFJvLeA}8CrkVO#1a2{>r{{B->{p)jbVUd%gRl{H&&BGpB(nUOv=RuGC4a)S{
z_7ehtCSQ;*$=_FbxN>lcnXybA+F%Od{s0mZ2y-%#HYn<8WfTVI)R7oDjO6JS5)VlW
z3ATt+OyDV|?=(V!vFwD_uP3r{$+fJS*ch>lkb<n2Gif6w^?*-gr=f6@Y;7R9kSceJ
zv}eQByIs>9Dn~kb78%|zY5^7=mBI{Asq7Do4IC&PgjNI%vFlJ#IbtL)jZ|-f$W;ZQ
zKHZ><Zqbl3cAIXy-xf$>+N24c-FyWruWWJvjsd<ZK9+fIlRO{SVhYyfwwdS8rhIg^
z66Lw5i#%5tIrTOY4A4siEmRoD8!$93#-7h&&&e)uiJ`?cS4qdf7(nEZ%~!M6$Jt*p
zCtbY-ODOm6nPA^CL@;QFfLE`Lnh_v`El@j#(Euzr+_#Ac&*h*CxP#TTB?E*rMDq6x
z>NYt*e7Lrx-U36QisPQ-W(s+{5Y$tXBZ5xvISqv+r8>wnr*qEVMmdsB(=yYutgP+!
zge#4CuKEyAdF;||Hy@P8QhX+>2dlFn)s^|<`!>`@fu>D_L_p-wB-I*hXnkk~M!kV-
z)Em{@+}U;C+}Z8;HdLUkg!(-$d*!(A47!vRyS}pI%!FOg)f+$vjlB)lVmD}uUZv_!
zASqCc&nq1-1%aS_x>#iivIW&h=XB^9f5kyO*}<%J-oLPO<?p@MHNBO6ok0Bo_OL!G
z*v-cezI0~qudW#DjrqhK-(lHWSxq45SXkOJp)dOqK?p=H^gY;5BNz#w9#O9LfWNRG
zs~oZBaQ0hZ3^9N+v=}Z#K+}>9gC$wOr6~O8&=>c?C-5RTm~Fhc1-j<gSgJ12i&61l
zJc6OOf%M_aubVCwWJ&(Pp81bfH34V}Fr#=Yu<pB-mX{p)HMu(~wZUt2?-Se(_YTSI
zE$82v2~z3q-BDfek=_I5T)z{lHfUQ`YwJ%q^bLPlmz9n2{>??HY(qiw8I_gPU<l=x
z&bFJE4ta<MV7LftQHADUQy9Y?2^Eih(P0}}s{-7=!1{00?@;v`hf7b2t{6eWyWr8d
zt`xfCKs_y>omJEE<V4*>nOA(XYZD_YT`R!zkwb>s$3y;%>k-(GO~*$3P&?<q6odnq
zf_B=lJ{-!zxW3|0<1+88R`1|89|Fme7g@Bv;ZLRd0N1c~v1(jb4kl1?V4dsRHY~?_
zP#<N{5iiZB^5mtUxmM>|aC~>wAomb@`duJCpf7|@>weJhw!N*_<Bd3q!x-ct-DayU
JywYgW{{y(4m*fBd

literal 0
HcmV?d00001

diff --git a/dataset/casia_hwdb.py b/dataset/casia_hwdb.py
index 08a5253..39bbb99 100755
--- a/dataset/casia_hwdb.py
+++ b/dataset/casia_hwdb.py
@@ -7,13 +7,17 @@ we using this class to get .png and label from raw
 
 """
 from alfred.dl.tf.common import mute_tf
-
 mute_tf()
 import struct
 import numpy as np
 import cv2
 import tensorflow as tf
 
+import os
+
+
+this_dir = os.path.dirname(os.path.abspath(__file__))
+
 
 class CASIAHWDBGNT(object):
     """
@@ -52,8 +56,9 @@ def parse_example(record):
                                                   tf.io.FixedLenFeature([], tf.string),
                                           })
     img = tf.io.decode_raw(features['image'], out_type=tf.uint8)
+    img = tf.cast(tf.reshape(img, (64, 64)), dtype=tf.float32)
     label = tf.cast(features['label'], tf.int32)
-    return img, label
+    return {'image': img, 'label': label}
 
 
 def load_ds():
@@ -63,14 +68,15 @@ def load_ds():
     return ds
 
 
-def load_charactors():
-    a = open('charactors.txt', 'r').readlines()
+def load_characters():
+
+    a = open(os.path.join(this_dir, 'charactors.txt'), 'r').readlines()
     return [i.strip() for i in a]
 
 
 if __name__ == "__main__":
     ds = load_ds()
-    charactors = load_charactors()
+    charactors = load_characters()
     for img, label in ds.take(9):
         # start training on model...
         img = img.numpy()
diff --git a/dataset/dataset_hwdb.py b/dataset/dataset_hwdb.py
deleted file mode 100755
index e69de29..0000000
diff --git a/models/__pycache__/cnn_net.cpython-36.pyc b/models/__pycache__/cnn_net.cpython-36.pyc
new file mode 100755
index 0000000000000000000000000000000000000000..9402b814e21052ef6a4159873ca63a6a73dd13d7
GIT binary patch
literal 2865
zcma)8TW=Fb6y9Cmof{3|()&uZN-SA1u_>x5j6mF8id89Zp=h-1jFUC%U3X>;7^M=5
z(~Br6Y7r_RE(MjA7AcemM3gk;f9z{xJAUeO+n1iRYddxXh_&X-%$b>U{pNTw0|W8P
zd;a-!F)m4eNnKwM#s{DYBvhJ`s7w`8o>HI>n1Nz&Dkw`dNJI0|REUOY1p4qXiAHH`
zUZOFk6e5LaH82&U%A}Ntm;VESarkuGIzuMZtSU^SIF7S49W#pBG!sd6@1CT3B9l}z
zCzI+#QY~qe8rF1nbn@WwBcn-G=yr)^NAb|aXd=E#9W$&_*%3*rTqJhc@x~`=3r`!r
zJgWb=w6^qH{p#;0)zgey%p}4%&ypLfjT`gLhu3P$KmYy5rOpvx))rS!3Wpn%WhauG
zEJ(zQ+Bs6PZF9qgHV54BeFGQLNi_{FwEJEu2|otuO)+T8A>jB#`ooPt^q%gCLK>pb
zIoA=`=CmyWYuc8y`(BwXi!v{=2;KQ$?<-l9>9(jNaI-~_;<hF2ezQgKys0@3v)VSb
zrekeUB$J5e^$w8oWz(z>NPxw3htV-7KW<s$l-ni1#aQcb5}?3Jq{Qu02?+A;G{8;M
z9pjAV7`8?7R@Oh}Xx2<8&Y%Q*GVN)@5glV*u1$v&c56EN@xf!0<NA!o1#^5HZql2x
zE!L$G^SVoYxz3JHuPU9hvS3+{cB%4w6X+TE&KWK1j1-w>wIV1=XfzRa-oeQ0+!lf`
z%YndF$XUagA-<JkewX#Tq-tuXm?^S;mq2eS>vtbCep-ebR=@VBws^aF{foxqmD<-2
z;r>;Kzu3GtUw`sVW93Ty_Dx`~KfCU4TwA)>ymb%G)h}MGeS4>V>pqBS+`rvex!hR&
zuCa1+{m(14ua;|%=YZd`wf6m`Zfn<8uhwopSX;WV{^aTU;@!sL51?ReX?bn=%k^ig
z^~ZC~3wNLx#5G+nb4{;2Hz2wu91INmtgM+h(D*cjrR6+83SiZ8If?Pio3`d;(%^p+
zs;HDp$ZpuwDvS$VK?ZyQXfbGqpotqm9BEdn%2YnBh!F>BZdIO>9i<Q`1gp~L%B)fd
zd9!dKGAq#l?WY6OtG^53sw_)g>rpyLV+-;oZmbH@d-laaN|?qMKvOS&R;GO$e5m3>
zxh9%>eur(D6;+f=C7U~HMg?TUy)#8qh1I4E#0{O%OoO^X&dhQm?Dlzunur;##N0^U
zFdc}2+t*c@-H2awg&S#=0vB@8zMD-97ep`S)n2UR37HR|qDU*<35BC=9)WQP$4Fp2
zh+`y)J~v=8%N;x@1mk$;k8o~tS8?)g2yD@W8=r*S0=^B+bOZ4B!%?vtK7{{Mhuzq5
z?c66&{b3k6>enYXbeLI!B?3H(J7X5nO%22gH&nDKGk=x%+pzHO{#21sLrWD56mF#~
z=4g6KGg5|S*?s{^**e!P;^fmQ@2E)WmPKIWc&XwJoGKe8#TnVXd*V$HDMo-u@`xOl
z`{gLK?eeg^6Mp<%;P(IwBjo{X4hD`?kPC_ffC2!?bck-jzi}E~2m>!fUxQh6!CR*>
zIt;V;Lcm+u+O^V0N4i$FbymVv3HC++mLWReNr?cA25~$ctI7yq0Mhnc^8i@lXAHCC
z28C8+JOaZ+*o!2`OTpn)BJn;X{YcOcKa&a{YIB<xU<C2sIDmVb65!B~;MLJW3g3d;
zUIit+dp6MXB7T%_1<@SS!M7nn%<vsZb|QHl$s0ggAmNxj{uUC1hELvsQE!sdW55z9
zL6XC*6c0iVO~LuF!ry~69XGT;)O-M%hYco8$?$T+7n=gFvj7c1=z{>}TrS5P9q%cA
z*kkhmg3mrG!9F?X*#&nlD43bwY-yAb!!jH~_Q3&xY9;ynkS}?>y_N%Qu+f6l65_@Q
z0a%w!9QPA~CDdQxqj0)Qffp2v2Y(;PN6<V4QN^1l;%<ZxYU?Nx9tgtbP-f41IhfkA
u!5H^0w4V#S5MHpnB)OsEcr!g|&r|D0_xXnI$CL`hU{HYz7K{!=hyMpOJ4#Xj

literal 0
HcmV?d00001

diff --git a/models/cnn_net.py b/models/cnn_net.py
index 8325011..f70460a 100755
--- a/models/cnn_net.py
+++ b/models/cnn_net.py
@@ -1,4 +1,3 @@
-
 '''
 
 
@@ -21,9 +20,42 @@ accuracy = tf.reduce_mean(tf.cast(tf.equal(tf.argmax(logits, 1), labels), tf.flo
 '''
 
 import tensorflow as tf
+from tensorflow.keras import layers
 
 
+# some simple models
+def build_net_001(input_shape, n_classes):
+    assert len(input_shape) == 3, 'only support 3 channels'
+    model = tf.keras.Sequential()
+    model.add(tf.keras.layers.Conv2D(
+        input_shape=input_shape, filters=32, kernel_size=(3, 3), strides=(1, 1),
+        padding='valid', activation='relu'))
+    model.add(tf.keras.layers.MaxPool2D(pool_size=(2, 2)))
+    model.add(tf.keras.layers.Flatten())
+    model.add(tf.keras.layers.Dense(32, activation='relu'))
+    model.add(tf.keras.layers.Dense(n_classes, activation='softmax'))
+    return model
+
+
+def build_net_002(input_shape, n_classes):
+    model = tf.keras.Sequential([
+        layers.Conv2D(input_shape=input_shape, filters=64, kernel_size=(3, 3), strides=(1, 1),
+                      padding='same', activation='relu'),
+        layers.MaxPool2D(pool_size=(2, 2), padding='same'),
+        layers.Conv2D(filters=128, kernel_size=(3, 3), padding='same'),
+        layers.MaxPool2D(pool_size=(2, 2), padding='same'),
+        layers.Conv2D(filters=256, kernel_size=(3, 3), padding='same'),
+        layers.MaxPool2D(pool_size=(2, 2), padding='same'),
+
+        layers.Flatten(),
+        layers.Dense(1024, activation='relu'),
+        layers.Dense(n_classes, activation='softmax')
+    ])
+    return model
+
+
+# some models wrapped into tf.keras.Model
 class CNNNet(tf.keras.Model):
 
-    def __init__(self.):
-        pass
\ No newline at end of file
+    def __init__(self):
+        pass
diff --git a/train.py b/train.py
index c4ffb29..f589719 100755
--- a/train.py
+++ b/train.py
@@ -3,6 +3,7 @@ training HWDB Chinese charactors classification
 on MobileNetV2 
 '''
 from alfred.dl.tf.common import mute_tf
+
 mute_tf()
 
 import os
@@ -12,40 +13,41 @@ import tensorflow as tf
 
 from alfred.utils.log import logger as logging
 import tensorflow_datasets as tfds
-from dataset.casia_hwdb import load_ds, load_charactors
-from models.cnn_net import CNNNet
+from dataset.casia_hwdb import load_ds, load_characters
+from models.cnn_net import CNNNet, build_net_002
 
 
-target_size = 224
+
+target_size = 64
 num_classes = 7356
-use_keras_fit = False
-# use_keras_fit = True
-ckpt_path = './checkpoints/no_finetune/flowers_mbv2_scratch-{epoch}.ckpt'
+# use_keras_fit = False
+use_keras_fit = True
+ckpt_path = './checkpoints/cn_ocr-{epoch}.ckpt'
 
 
 def preprocess(x):
     """
     minus mean pixel or normalize?
     """
+    x['image'] = tf.expand_dims(x['image'], axis=-1)
     x['image'] = tf.image.resize(x['image'], (target_size, target_size))
     x['image'] /= 255.
-    x['image'] = 2*x['image'] - 1
+    x['image'] = 2 * x['image'] - 1
     return x['image'], x['label']
 
+
 def train():
-    all_charactors = load_charactors()
-    num_classes = len(all_charactors)
-    # using mobilenetv2 classify tf_flowers dataset
+    all_characters = load_characters()
+    num_classes = len(all_characters)
+    logging.info('all characters: {}'.format(num_classes))
     train_dataset = load_ds()
     train_dataset = train_dataset.shuffle(100).map(preprocess).batch(4).repeat()
 
     # init model
-    model = CNNNet()
-
-    # model.summary()
-    # model = tf.keras.models.load_model('flowers_mobilenetv2.h5')
+    model = build_net_002((64, 64, 1), num_classes)
+    model.summary()
     logging.info('model loaded.')
-    
+
     start_epoch = 0
     latest_ckpt = tf.train.latest_checkpoint(os.path.dirname(ckpt_path))
     if latest_ckpt:
@@ -56,26 +58,24 @@ def train():
         logging.info('passing resume since weights not there. training from scratch')
 
     if use_keras_fit:
-        # todo: why keras fit converge faster than tf loop?
         model.compile(
-            optimizer='adam',
-            loss='sparse_categorical_crossentropy',
+            optimizer=tf.keras.optimizers.Adam(),
+            loss=tf.keras.losses.SparseCategoricalCrossentropy(),
             metrics=['accuracy'])
         try:
             model.fit(
-            train_dataset, epochs=50,             
-            steps_per_epoch=700,)
+                train_dataset, epochs=50,
+                steps_per_epoch=700, )
         except KeyboardInterrupt:
             model.save_weights(ckpt_path.format(epoch=0))
             logging.info('keras model saved.')
         model.save_weights(ckpt_path.format(epoch=0))
-        model.save(os.path.join(os.path.dirname(ckpt_path), 'flowers_mobilenetv2.h5'))
+        model.save(os.path.join(os.path.dirname(ckpt_path), 'cn_ocr.h5'))
     else:
         loss_fn = tf.losses.SparseCategoricalCrossentropy()
         optimizer = tf.optimizers.RMSprop()
 
         train_loss = tf.metrics.Mean(name='train_loss')
-        # the accuracy calculation has some problems, seems not right?
         train_accuracy = tf.metrics.SparseCategoricalAccuracy(name='train_accuracy')
 
         for epoch in range(start_epoch, 120):
@@ -92,7 +92,7 @@ def train():
                     train_accuracy(labels, predictions)
                     if batch % 10 == 0:
                         logging.info('Epoch: {}, iter: {}, loss: {}, train_acc: {}'.format(
-                        epoch, batch, train_loss.result(), train_accuracy.result()))
+                            epoch, batch, train_loss.result(), train_accuracy.result()))
             except KeyboardInterrupt:
                 logging.info('interrupted.')
                 model.save_weights(ckpt_path.format(epoch=epoch))
@@ -100,7 +100,5 @@ def train():
                 exit(0)
 
 
-
 if __name__ == "__main__":
     train()
-