From 18659b7d68782cdeb9e429ff742acaf1eec2b2b7 Mon Sep 17 00:00:00 2001
From: hua <hsj@mys4s.cn>
Date: Sun, 25 Nov 2018 14:56:04 +0800
Subject: [PATCH] bug fix

---
 .idea/workspace.xml                   |  64 ++++++-----
 lib/__pycache__/common.cpython-36.pyc | Bin 6705 -> 6704 bytes
 lib/common.py                         |   2 +-
 lib/gen/generator.py                  |  97 ++++++-----------
 readme.md                             |   2 +
 src/net_work.py                       |   2 -
 src/util.py                           | 147 --------------------------
 7 files changed, 69 insertions(+), 245 deletions(-)
 delete mode 100644 src/util.py

diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 6d2bff6..8914d6d 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -1,7 +1,9 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
   <component name="ChangeListManager">
-    <list default="true" id="497e7824-1d53-428b-a3ae-e667d9334ad4" name="Default" comment="" />
+    <list default="true" id="497e7824-1d53-428b-a3ae-e667d9334ad4" name="Default" comment="">
+      <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
+    </list>
     <option name="EXCLUDED_CONVERTED_TO_IGNORED" value="true" />
     <option name="TRACKING_ENABLED" value="true" />
     <option name="SHOW_DIALOG" value="false" />
@@ -14,11 +16,11 @@
   </component>
   <component name="FileEditorManager">
     <leaf>
-      <file leaf-file-name="train_vin_code.py" pinned="false" current-in-tab="true">
+      <file leaf-file-name="train_vin_code.py" pinned="false" current-in-tab="false">
         <entry file="file://$PROJECT_DIR$/src/train_vin_code.py">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="198">
-              <caret line="9" column="14" lean-forward="true" selection-start-line="9" selection-start-column="14" selection-end-line="9" selection-end-column="14" />
+            <state relative-caret-position="418">
+              <caret line="19" lean-forward="true" selection-start-line="19" selection-end-line="19" />
               <folding>
                 <element signature="e#47#81#0" expanded="true" />
               </folding>
@@ -26,6 +28,15 @@
           </provider>
         </entry>
       </file>
+      <file leaf-file-name="readme.md" pinned="false" current-in-tab="true">
+        <entry file="file://$PROJECT_DIR$/readme.md">
+          <provider selected="true" editor-type-id="text-editor">
+            <state relative-caret-position="66">
+              <caret line="3" selection-start-line="3" selection-end-line="3" />
+            </state>
+          </provider>
+        </entry>
+      </file>
       <file leaf-file-name="common.py" pinned="false" current-in-tab="false">
         <entry file="file://$PROJECT_DIR$/lib/common.py">
           <provider selected="true" editor-type-id="text-editor">
@@ -56,6 +67,14 @@
       </list>
     </option>
   </component>
+  <component name="FindInProjectRecents">
+    <findStrings>
+      <find>print</find>
+    </findStrings>
+  </component>
+  <component name="Git.Settings">
+    <option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$" />
+  </component>
   <component name="IdeDocumentHistory">
     <option name="CHANGED_PATHS">
       <list>
@@ -66,6 +85,7 @@
         <option value="$PROJECT_DIR$/src/net_work.py" />
         <option value="$PROJECT_DIR$/lib/gen/generator.py" />
         <option value="$PROJECT_DIR$/src/train_vin_code.py" />
+        <option value="$PROJECT_DIR$/readme.md" />
       </list>
     </option>
   </component>
@@ -90,6 +110,7 @@
     <option name="width" value="954" />
     <option name="height" value="1080" />
   </component>
+  <component name="ProjectLevelVcsManager" settingsEditedManually="true" />
   <component name="ProjectView">
     <navigator proportions="" version="1">
       <foldersAlwaysOnTop value="true" />
@@ -103,27 +124,6 @@
               <item name="vin_recognize" type="b2602c69:ProjectViewProjectNode" />
               <item name="vin_recognize" type="462c0819:PsiDirectoryNode" />
             </path>
-            <path>
-              <item name="vin_recognize" type="b2602c69:ProjectViewProjectNode" />
-              <item name="vin_recognize" type="462c0819:PsiDirectoryNode" />
-              <item name="lib" type="462c0819:PsiDirectoryNode" />
-            </path>
-            <path>
-              <item name="vin_recognize" type="b2602c69:ProjectViewProjectNode" />
-              <item name="vin_recognize" type="462c0819:PsiDirectoryNode" />
-              <item name="lib" type="462c0819:PsiDirectoryNode" />
-              <item name="gen" type="462c0819:PsiDirectoryNode" />
-            </path>
-            <path>
-              <item name="vin_recognize" type="b2602c69:ProjectViewProjectNode" />
-              <item name="vin_recognize" type="462c0819:PsiDirectoryNode" />
-              <item name="src" type="462c0819:PsiDirectoryNode" />
-            </path>
-            <path>
-              <item name="vin_recognize" type="b2602c69:ProjectViewProjectNode" />
-              <item name="vin_recognize" type="462c0819:PsiDirectoryNode" />
-              <item name="static" type="462c0819:PsiDirectoryNode" />
-            </path>
           </expand>
           <select />
         </subPane>
@@ -198,12 +198,13 @@
   </component>
   <component name="ToolWindowManager">
     <frame x="1440" y="-180" width="1920" height="1080" extended-state="6" />
+    <editor active="true" />
     <layout>
       <window_info anchor="bottom" id="TODO" order="6" />
       <window_info anchor="bottom" id="Event Log" side_tool="true" />
       <window_info anchor="bottom" id="Database Changes" show_stripe_button="false" />
       <window_info anchor="bottom" id="Run" order="2" weight="0.63441956" />
-      <window_info anchor="bottom" id="Version Control" show_stripe_button="false" />
+      <window_info anchor="bottom" id="Version Control" />
       <window_info anchor="bottom" id="Python Console" />
       <window_info active="true" anchor="bottom" id="Terminal" visible="true" weight="0.3299389" />
       <window_info content_ui="combo" id="Project" order="0" visible="true" weight="0.15921193" />
@@ -311,14 +312,21 @@
     </entry>
     <entry file="file://$PROJECT_DIR$/src/train_vin_code.py">
       <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="198">
-          <caret line="9" column="14" lean-forward="true" selection-start-line="9" selection-start-column="14" selection-end-line="9" selection-end-column="14" />
+        <state relative-caret-position="418">
+          <caret line="19" lean-forward="true" selection-start-line="19" selection-end-line="19" />
           <folding>
             <element signature="e#47#81#0" expanded="true" />
           </folding>
         </state>
       </provider>
     </entry>
+    <entry file="file://$PROJECT_DIR$/readme.md">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="66">
+          <caret line="3" selection-start-line="3" selection-end-line="3" />
+        </state>
+      </provider>
+    </entry>
   </component>
   <component name="masterDetails">
     <states>
diff --git a/lib/__pycache__/common.cpython-36.pyc b/lib/__pycache__/common.cpython-36.pyc
index ebfca0d6b8e3cf1ea69a0b8bdefe9659204c3cab..c7fed52b2cdb02bea8c433720da2b8aaa14f2edc 100644
GIT binary patch
delta 731
zcmZ{i&u<b@5XTv4ksq#Qk@CZ;A*qEdDrqZ9t<};NX{=UF4+bI4f`~|h&^&0*J!(x)
zbI_Z5>cN9;xOmV+PxjcDn3x_+JfMkxf(PS#A*m-f+0UCdZ)U!;^L8h(op7GhG?!=o
zcmCs!vra>X#bR-V<@xc1JDc0#&b#%_*3)pUQE9Z9PB|#YZ=G2hY3r`9L`M6UdyXi<
zaZOQ-Uuh?XFGvBE6c{!YgM9$QstAq={;mb=lifJ}qpi}Mq=$I(NNV9coV}BS)NW2h
zwg#z-Sgijd{;LrYjlroy%8D|V02?p`Tn7BW6@k&OreUW54=@AF^232RrTN*w3mxLh
zDG=$yvgCN)GeH?%_e?vYaFa^$C(o8lprIe5C~xRP_OnQe^K1RMjBHoGMvLtAz8CI!
z{dApwdSkTI_8uK4ittrmT7ouw3zTht@g-?8XII7h)zzlTqr9!B>Qv=Yt5~np$~k+f
zs!Fx8^xp0GF0^t$o-_U#TIR?8H}2DLpkH-{X9G$1D9kZnocDAeKMzEyz#jr@qd9kD
zt*mOrm6d9}qWTfig&meD_$+U;PhhcnNtyg0SO{V&`+co)vn37Ql#RLt+~#k=K&UFF
zOsfdby`J5@N5Z}EaA@qt2#)Z#sRS?wq=5|J0v3TKAPX!51pqDWJ4~})d{j0~`Mj5*
Oldd~B85GF4L)E`&0G#Ro

delta 712
zcmZ|N%WD&H5C`x~(#AYon~iB6S|y+~O=_#Qjj3(5jY*^yg<2{2sIDPy*Fa3dFKMp|
z9wew22fWr>!HX?9cu@oo_9%!4SrELW;J+XU&No6mxv-y|o!?_-f9%%8_Jns_(|qaN
zuUj9ty;T}WIUEjOT&~>~cb;#>JMSKKHlM~THLGT|U(j(6+5Fafod(+h-#4OE`=>ug
zl;#ml(J;T#jvUL#4r)?Jy3{b@J(5mENSx&#TEsowoyWhmHOkBK7(Y8Ow{Ql@zax7~
zyK55inj|%ahx6Zz|JF!IjsBT^%86}efD4=h=Rp|E3dvqIhd2uY-~!08)i+87e%ZI7
zqqy=4QhL8UY`z$nq<MZAnDq=HO)ATu0-JIIRs9eRaZ5k!9>t=Q{8}H7Biqq0(IV@?
z_u}VZn6B_{Fh$F4{ooi;g7cxgOlpM|Xr=u%l%Xlxt%~=`YYk;!dsnyWO|LRb&0^K6
zn6|rAR;7w5y>~mljaqkr!G&;^O8g}J#(x3{^s7cV8_D=jBOC+c{5UdAtNbCde#-XO
z)=gC@*6Zb}rFt>k#f``mzR2C^5gg7SDU;3U-6&4P?rEBhrZjg|Uepq}#^0k6tuCRQ
zl!eTFrr6TuL4^3VR2t+#0n7s*SOm*p1(d)lKnu3VFsj7|reVmZY{ZWGZeXMzBfJwU
F{{gafoxuPA

diff --git a/lib/common.py b/lib/common.py
index 50f14a4..9562151 100644
--- a/lib/common.py
+++ b/lib/common.py
@@ -8,7 +8,7 @@
 
 def print_start():
     """
-    打印 开始 logog
+    打印 开始 logo
     :return:
     """
 
diff --git a/lib/gen/generator.py b/lib/gen/generator.py
index 0feaf92..09efc06 100644
--- a/lib/gen/generator.py
+++ b/lib/gen/generator.py
@@ -13,17 +13,19 @@
 from lib.gen.static import *
 
 # 路径先采用绝对路径
+test_gen_path = '/Volumes/doc/projects/ml/vin_recognize/static/test/'
 images_path = "/Volumes/doc/projects/ml/vin_recognize/static/images/"
 font_path = '/Volumes/doc/projects/ml/vin_recognize/static/font/'
 bad_path = '/Volumes/doc/projects/ml/vin_recognize/static/bad/'
 predict_path = '/Volumes/doc/projects/ml/vin_recognize/static/predict/'
 font = 'OCR-B.ttf'
 
+
 class GenVin:
 
-    def __init__(self, height, width,predict_path=predict_path,fontSize=30,
-                 NoPlates=bad_path,font_path=font_path,
-                 images_path=images_path,font=font):
+    def __init__(self, height, width, predict_path=predict_path, fontSize=30,
+                 NoPlates=bad_path, font_path=font_path,
+                 images_path=images_path, font=font):
 
         self.fontSize = fontSize
         self.font = ImageFont.truetype(font_path + font, self.fontSize, 0)
@@ -56,11 +58,12 @@ def draw(self, val):
         :param val:
         :return:
         """
-        height,width =  self.bg.shape
+        height, width = self.bg.shape
         offset = 0
-        base = int((width - self.max_size*self.fontSize)/2)
+        base = int((width - self.max_size * self.fontSize) / 2)
         for i, item in enumerate(val):
-            self.bg[int((height-self.fontSize)/2):int((height+self.fontSize)/2), base: base + self.fontSize] = self.draw_item(item, 0, 0)
+            self.bg[int((height - self.fontSize) / 2):int((height + self.fontSize) / 2),
+            base: base + self.fontSize] = self.draw_item(item, 0, 0)
             base += self.fontSize + offset
         return self.bg
 
@@ -86,13 +89,13 @@ def GenCh(self, val, img):
         """
         img = Image.new("RGB", (self.fontSize, self.bg.shape[1]), (0, 0, 0))
         draw = ImageDraw.Draw(img)
-        draw.text((0, int(self.bg.shape[1]-self.fontSize)/2), val, (255, 255), font=self.font)
+        draw.text((0, int(self.bg.shape[1] - self.fontSize) / 2), val, (255, 255), font=self.font)
         A = np.array(img)
         return A
 
-    def random_start(self,seed):
+    def random_start(self, seed):
         ran = r(seed)
-        return ran%3 == 0
+        return ran % 3 == 0
 
     def generate(self, text):
         fg = self.draw(text)
@@ -125,7 +128,7 @@ def gen_sample(self):
         return label, img[:, :, 2]  # 返回的label为标签，img为深度为3的图像像素
 
     def gen_img(self):
-        text, label = self.get_text_v2()
+        text, label = self.get_text()
         img = self.generate(text)
         img = cv2.resize(img, (self.width, self.height))
         img = np.multiply(img, 1 / 255.0)  # [height,width,channel]
@@ -135,13 +138,13 @@ def gen_img(self):
     def get_next_batch(self, batch_size=128):
         batch_x = np.zeros([batch_size, self.width * self.height])
         batch_y = np.zeros([batch_size, self.len * self.max_size])
-        labels  = []
+        labels = []
         for i in range(batch_size):
             image, text, vec = self.gen_img()
             labels.append(text)
             batch_x[i, :] = image.reshape((self.width * self.height))
             batch_y[i, :] = vec
-        return batch_x, batch_y,labels
+        return batch_x, batch_y, labels
 
     def get_predict_source(self):
         items = os.listdir(self.predict_path)
@@ -150,17 +153,16 @@ def get_predict_source(self):
         textes = []
         for i, item in enumerate(items):
             textes.append(item.split("/")[-1])
-            img = cv2.imread(os.path.join(path,item), cv2.IMREAD_GRAYSCALE)
-            img = cv2.resize(img,(self.width,self.height))
+            img = cv2.imread(os.path.join(self.predict_path, item), cv2.IMREAD_GRAYSCALE)
+            img = cv2.resize(img, (self.width, self.height))
             batch_x[i, :] = img.reshape((self.width * self.height))
 
-        return batch_x,textes
+        return batch_x, textes
 
-
-    def text2vec(self,text):
+    def text2vec(self, text):
         text_len = len(text)
         if text_len > self.max_size:
-            raise ValueError('字符最长 ',self.max_size,'个字符')
+            raise ValueError('字符最长 ', self.max_size, '个字符')
         vector = np.zeros(self.max_size * self.len)
         for i, c in enumerate(text):
             idx = i * self.len + chars_indexes[c]
@@ -168,7 +170,7 @@ def text2vec(self,text):
         return vector
 
     # 向量转回文本
-    def vec2text(self,vec):
+    def vec2text(self, vec):
         char_pos = vec.nonzero()[0]
         text = []
         for i, c in enumerate(char_pos):
@@ -176,43 +178,19 @@ def vec2text(self,vec):
             text.append(chars[char_idx])
         return "".join(text)
 
-    def genBatch(self, batchSize, outputPath):
-        if (not os.path.exists(outputPath)):
-            os.mkdir(outputPath)
-        for i in range(batchSize):
-            # plateStr = self.genPlateString(-1, -1)
-            plateStr, label = self.get_text()
-            img = self.generate(plateStr)
-            img = cv2.resize(img, (self.width, self.height))
-            filename = os.path.join(outputPath, str(i).zfill(4) + '.' + "".join(plateStr) + ".jpg")
-            cv2.imwrite(filename, img)
-
-    def genBatch_v2(self, batchSize, outputPath):
+    def gen_Batch(self, batchSize, outputPath):
         if (not os.path.exists(outputPath)):
             os.mkdir(outputPath)
 
         for i in range(batchSize):
             # plateStr = self.genPlateString(-1, -1)
-            plateStr, label = self.get_text_v2()
+            plateStr, label = self.get_text()
             img = self.generate(plateStr)
             img = cv2.resize(img, (self.width, self.height))
             filename = os.path.join(outputPath, str(i).zfill(4) + '.' + "".join(plateStr) + ".png")
             cv2.imwrite(filename, img)
 
     def get_text(self):
-        text = []
-        label = []
-        text += (self.get_random_char(STATE_CODE, 1))  # 生成车牌
-        text += (self.get_random_char(chars, 2))
-        text += (self.get_random_char(FOUR_CODE, 1))  # 4
-        text += (self.get_random_char(chars, 4))  # 8
-        text += (self.get_random_char(NINE_CODE, 1))  # 9
-        text += (self.get_random_char(chars, 8))
-        for item in text:
-            label.append(chars_indexes[item])
-        return ''.join(text), np.array(label)
-
-    def get_text_v2(self):
         """
          # 随机生成字串，长度固定
         # 返回text,及对应的向量
@@ -226,11 +204,6 @@ def get_text_v2(self):
         text += (self.get_random_char(chars, 4))  # 8
         text += (self.get_random_char(NINE_CODE, 1))  # 9
         text += (self.get_random_char(chars, 8))
-        # size = random.randint(1, self.max_size)
-        # size = self.max_size
-        # for i, c in enumerate(text):
-        #     vec = self.text2vec(c)
-        #     vecs[i * self.len:(i + 1) * self.len] = np.copy(vec)
         return text, self.text2vec(text)
 
     def get_random_char(self, chars, size):
@@ -240,21 +213,11 @@ def get_random_char(self, chars, size):
         for i in ints: result.append(chars[i])
         return result
 
-# generator = GenVin("font.ttf", 49, 258, fontSize=30,NoPlates=bad_path)
-# if __name__ == '__main__':
-#     generator = GenVin("./font/DejaVuSansMono.ttf",80,440)
-#     print("".join(generator.get_text()))
-#     generator = GenVin("./font/Bitter-Regular.ttf")
-#     generator = GenVin("./font/SimHei.ttf")
-#     label,img = generator.gen_sample()
-#     print(label)d
-#     cv2.imshow("test",img)
-#     cv2.waitKeyEx(0)
-#     generator.genBatch_v2(3, "../../data/train/energy")
-#     path = "/Volumes/doc/projects/ml/ocr_tensorflow_cnn/data/train/energy/"
-#     for item in os.listdir(path):
-#         img = cv2.imread(os.path.join(path,item), cv2.IMREAD_GRAYSCALE)
-#         print(img.shape)
-#         cv2.imshow("123", img)
-#         cv2.waitKey(0)
 
+generator = GenVin(49, 258)
+if __name__ == '__main__':
+    # generator = GenVin("./font/DejaVuSansMono.ttf",80,440)
+    # print("".join(generator.get_text()))
+    # generator = GenVin("./font/Bitter-Regular.ttf")
+    # generator = GenVin("./font/SimHei.ttf")
+    generator.gen_Batch(3, test_gen_path)
\ No newline at end of file
diff --git a/readme.md b/readme.md
index 7cc62df..a3ea764 100644
--- a/readme.md
+++ b/readme.md
@@ -1,3 +1,5 @@
 # vin recognize
 > 车辆 vin 码识别
 
+## 关键在于样本的生成
+
diff --git a/src/net_work.py b/src/net_work.py
index e2a9b62..5e7b3ad 100644
--- a/src/net_work.py
+++ b/src/net_work.py
@@ -6,8 +6,6 @@
 import tensorflow as tf
 from lib.gen.generator import GenVin
 import numpy as np
-from lib.logger import logger
-
 
 class NetWork():
 
diff --git a/src/util.py b/src/util.py
deleted file mode 100644
index 0a3dc99..0000000
--- a/src/util.py
+++ /dev/null
@@ -1,147 +0,0 @@
-import cv2
-import numpy as np
-import tensorflow as tf
-
-from lib.gen.static import chars
-from lib.gen.generator import generator
-
-# 分类类别为
-num_classes = len(chars) + 1  # len(chars)  + ctc blank
-
-# ===========  tensorflow 常量 =========== #
-
-"""
-
-Notice : 训练车牌和训练 vin 码，采用同一个 util 文件
-
-使用前要进行修改
-
-"""
-
-tf.flags.DEFINE_boolean('restore', False, 'whether to restore from the latest checkpoint')
-tf.flags.DEFINE_string('checkpoint_dir', './checkpoint/', 'the checkpoint dir')
-tf.flags.DEFINE_float('initial_learning_rate', 1e-3, 'initial lr')
-
-tf.flags.DEFINE_integer('image_height', 80, 'image height')
-tf.flags.DEFINE_integer('image_width', 440, 'image width')
-tf.flags.DEFINE_integer('image_channel', 1, 'image channels as input')
-
-tf.flags.DEFINE_integer('cnn_count', 4, 'count of cnn module to extract image features.')
-tf.flags.DEFINE_integer('out_channels', 64, 'output channels of last layer in CNN')
-tf.flags.DEFINE_integer('num_hidden', 128, 'number of hidden units in lstm')
-tf.flags.DEFINE_float('output_keep_prob', 0.8, 'output_keep_prob in lstm')
-tf.flags.DEFINE_integer('num_epochs', 10000, 'maximum epochs')
-tf.flags.DEFINE_integer('batch_size', 128, 'the batch_size')
-tf.flags.DEFINE_integer('batch', 10, 'the batch_size')
-tf.flags.DEFINE_integer('save_steps', 1000, 'the step to save checkpoint')
-tf.flags.DEFINE_float('leakiness', 0.01, 'leakiness of lrelu')
-tf.flags.DEFINE_integer('validation_steps', 500, 'the step to validation')
-
-tf.flags.DEFINE_float('decay_rate', 0.98, 'the lr decay rate')
-tf.flags.DEFINE_float('beta1', 0.9, 'parameter of adam optimizer beta1')
-tf.flags.DEFINE_float('beta2', 0.999, 'adam parameter beta2')
-
-tf.flags.DEFINE_integer('decay_steps', 10000, 'the lr decay_step for optimizer')
-tf.flags.DEFINE_float('momentum', 0.9, 'the momentum')
-
-tf.flags.DEFINE_string('train_dir', './imgs/train/', 'the train data dir')
-tf.flags.DEFINE_string('val_dir', './imgs/val/', 'the val data dir')
-tf.flags.DEFINE_string('infer_dir', './imgs/infer/', 'the infer data dir')
-tf.flags.DEFINE_string('log_dir', './log', 'the logging dir')
-tf.flags.DEFINE_string('mode', 'train', 'train, val or infer')
-tf.flags.DEFINE_integer('num_gpus', 0, 'num of gpus')
-FLAGS = tf.flags.FLAGS
-
-next_train_img_path = "/Volumes/doc/projects/ml/mtcnn_Tensorflow/test_data/next_train/img"
-next_train_img_txt = "train_img.txt"
-
-
-# 转化一个序列列表为稀疏矩阵
-def sparse_tuple_from(sequences, dtype=np.int32):
-    """
-    Create a sparse representention of x.
-    Args:
-        sequences: a list of lists of type dtype where each element is a sequence
-    Returns:
-        A tuple with (indices, values, shape)
-    """
-    indices = []
-    values = []
-
-    for n, seq in enumerate(sequences):
-        indices.extend(zip([n] * len(seq), range(len(seq))))
-        values.extend(seq)
-
-    indices = np.asarray(indices, dtype=np.int64)
-    values = np.asarray(values, dtype=dtype)
-    shape = np.asarray([len(sequences), np.asarray(indices).max(0)[1] + 1], dtype=np.int64)
-
-    return indices, values, shape
-
-
-def decode_sparse_tensor(sparse_tensor):
-    # print("sparse_tensor = ", sparse_tensor)
-    decoded_indexes = list()
-    current_i = 0
-    current_seq = []
-    for offset, i_and_index in enumerate(sparse_tensor[0]):
-        i = i_and_index[0]
-        if i != current_i:
-            decoded_indexes.append(current_seq)
-            current_i = i
-            current_seq = list()
-        current_seq.append(offset)
-    decoded_indexes.append(current_seq)
-    # print("decoded_indexes = ", decoded_indexes)
-    result = []
-    for index in decoded_indexes:
-        # print("index = ", index)
-        result.append(decode_a_seq(index, sparse_tensor))
-        # print(result)
-    return result
-
-
-# 解码
-def decode_a_seq(indexes, spars_tensor):
-    decoded = []
-    for m in indexes:
-        str = chars[spars_tensor[1][m]]
-        decoded.append(str)
-    # Replacing blank label to none
-    # str_decoded = str_decoded.replace(chr(ord('9') + 1), '')
-    # Replacing space label to space
-    # str_decoded = str_decoded.replace(chr(ord('0') - 1), ' ')
-    # print("ffffffff", str_decoded)
-    return decoded
-
-
-# 生成一个训练batch
-def get_next_batch(batch_size=128,ocriter=generator):
-    # (batch_size,256,32)
-    # FLAGS.image_height, FLAGS.image_width, FLAGS.image_channel
-    inputs = np.zeros([batch_size, FLAGS.image_width, FLAGS.image_height])
-    codes = []
-    for i in range(batch_size):
-        # 生成不定长度的字串
-        text, image = ocriter.gen_sample()
-        # print(image.shape,inputs.shape)
-        # np.transpose 矩阵转置 (32*256,) => (32,256) => (256,32)
-        inputs[i, :] = np.transpose(image.reshape((FLAGS.image_height, FLAGS.image_width)))
-        # inputs[i,:] = image
-        codes.append(list(text))
-    targets = [np.asarray(i) for i in codes]
-    # print(targets)
-    sparse_targets = sparse_tuple_from(targets)
-    # (batch_size,) 值都是256
-    seq_len = np.ones(inputs.shape[0]) * FLAGS.image_width
-    return inputs, sparse_targets, seq_len
-
-
-def get_next_batch_v2(batch_size=128,ocriter=generator):
-    return ocriter.get_next_batch_v2(batch_size)
-
-if __name__=="__main__":
-    # img = cv2.imread("/Volumes/doc/projects/ml/ocr_tensorflow_cnn/data/train/energy/Jietu20181120-205011.jpg",cv2.IMREAD_GRAYSCALE)
-    # cv2.imshow("test",img)
-    # cv2.waitKey(0)
-    print(np.ceil((3,3)))