A-Simple-Chinese-License-Plate-Generator-and-Recognition-Framework
A-Simple-Chinese-License-Plate-Generator-and-Recognition-Framework copied to clipboard
LCorleone
test_nn.py输出结果错误
想请教一下，我运行create_traindata生成了5000张图片，batch_size是256训练了20轮，但是在运行test_nn.py文件输出的结果却是完全错误的，想请教一下大家有遇到这种情况吗
test_nn.py代码如下： `import numpy as np import pickle from keras.layers import Dense, Input, BatchNormalization, Conv2D, Flatten, MaxPooling2D, Activation, Reshape, Layer from keras.models import Model, load_model, model_from_json from keras.callbacks import TensorBoard, ModelCheckpoint, LearningRateScheduler from keras.optimizers import RMSprop from keras import backend as K import tensorflow as tf import keras.backend.tensorflow_backend as KTF
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.4) sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
chars = ["京", "沪", "津", "渝", "冀", "晋", "蒙", "辽", "吉", "黑", "苏", "浙", "皖", "闽", "赣", "鲁", "豫", "鄂", "湘", "粤", "桂", "琼", "川", "贵", "云", "藏", "陕", "甘", "青", "宁", "新", "0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "A", "B", "C", "D", "E", "F", "G", "H", "J", "K", "L", "M", "N", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z"]
class NormLayer(Layer):
def init(self, **kwargs): super(NormLayer, self).init(**kwargs)
def build(self, input_shape): self.kernal = self.add_weight(name='NormLayer', shape=( 1, 13), initializer='ones', trainable=True) super(NormLayer, self).build(input_shape)
def call(self, inputs): # out = self.kernal * inputs out = K.dot(self.kernal, inputs) out = K.permute_dimensions(out, (1, 0, 2)) return out[:, 0, :] ''' because this NormLayer do not change the input_shape, so the compute_output_shape need not to implement (maybe) '''
def compute_output_shape(self, input_shape): return (input_shape[0], 65)
def get_config(self): # config = {} base_config = super(NormLayer, self).get_config() return dict(list(base_config.items())) # return dict(list(base_config.items()) + list(config.items())) e2e_model = load_model('e2e_model.h5', custom_objects={'NormLayer': NormLayer}) label_path = 'label.txt' tem_label = np.loadtxt(label_path) row, col = tem_label.shape label1 = np.zeros([row, 31]) label2 = np.zeros([row, 34]) label3 = np.zeros([row, 34]) label4 = np.zeros([row, 34]) label5 = np.zeros([row, 34]) label6 = np.zeros([row, 34]) label7 = np.zeros([row, 34]) for i in range(row): label1[i, int(tem_label[i, 0])] = 1 label2[i, int(tem_label[i, 1]) - 31] = 1 label3[i, int(tem_label[i, 2]) - 31] = 1 label4[i, int(tem_label[i, 3]) - 31] = 1 label5[i, int(tem_label[i, 4]) - 31] = 1 label6[i, int(tem_label[i, 5]) - 31] = 1 label7[i, int(tem_label[i, 6]) - 31] = 1
img_path = './train_data.pkl' img_data = np.load(img_path,allow_pickle=True) img_data = img_data.transpose(0, 2, 1, 3)
e2e_predict = e2e_model.predict(img_data[0:5, :, :, :])
def print_trueLabel(num): print(np.array([np.argmax(label1[num, :]), np.argmax(label2[num, :]) + 31, np.argmax(label3[num, :]) + 31, np.argmax(label4[num, :]) + 31, np.argmax(label5[num, :]) + 31, np.argmax(label6[num, :]) + 31, np.argmax(label7[num, :]) + 31])) print(chars[np.argmax(label1[num, :])] + chars[np.argmax(label2[num, :]) + 31] + chars[np.argmax(label3[num, :]) + 31] + chars[np.argmax(label4[num, :]) + 31] + chars[np.argmax(label5[num, :]) + 31] + chars[np.argmax(label6[num, :]) + 31] + chars[np.argmax(label7[num, :]) + 31])
def print_predictLabel(x): num = x[0].shape[0] sort = np.zeros([num, len(x)]) for i in range(len(x)): temp = x[i] sort[:,i] = np.argmax(temp, 1) for i in range(num): print(np.array([int(sort[i, 0]), int(sort[i, 1] + 31), int(sort[i, 2] + 31), int(sort[i, 3] + 31), int(sort[i, 4] + 31), int(sort[i, 5] + 31), int(sort[i, 6] + 31)])) print(chars[int(sort[i, 0])] + chars[int(sort[i, 1] + 31)] + chars[int(sort[i, 2] + 31)] + chars[int(sort[i, 3] + 31)] + chars[int(sort[i, 4] + 31)] + chars[int(sort[i, 5] + 31)] + chars[int(sort[i, 6] + 31)])
for i in range(5): print_trueLabel(i) print_predictLabel(e2e_predict)`
想请教一下，我运行create_traindata生成了5000张图片，batch_size是256训练了20轮，但是在运行test_nn.py文件输出的结果却是完全错误的，想请教一下大家有遇到这种情况吗
Hi brother, did you solve the problem? I encountered same issue
Yeah,because the time you run is too short. I run the code for a whole night,the recognized result was fine.
| | buptwanghao | | 邮箱：[email protected] |
Signature is customized by Netease Mail Master
在2020年03月07日 19:04，Inomjon 写道：
想请教一下，我运行create_traindata生成了5000张图片，batch_size是256训练了20轮，但是在运行test_nn.py文件输出的结果却是完全错误的，想请教一下大家有遇到这种情况吗
Hi brother, did you solve the problem? I encountered same issue
— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.
Yeah,because the time you run is too short. I run the code for a whole night,the recognized result was fine. | | buptwanghao | | 邮箱：[email protected] | Signature is customized by Netease Mail Master 在2020年03月07日 19:04，Inomjon 写道： 想请教一下，我运行create_traindata生成了5000张图片，batch_size是256训练了20轮，但是在运行test_nn.py文件输出的结果却是完全错误的，想请教一下大家有遇到这种情况吗 Hi brother, did you solve the problem? I encountered same issue — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.
thanks for your reply, do you mean training process? I generated 10000 images, what should I set for epoch? How many epochs should I run?
Yeah,train the model as long as possible.You can try to run the code before you sleep,and check the performance when you get up.
| | buptwanghao | | 邮箱：[email protected] |
Signature is customized by Netease Mail Master
在2020年03月09日 15:15，Inomjon 写道：
Yeah,because the time you run is too short. I run the code for a whole night,the recognized result was fine. | | buptwanghao | | 邮箱：[email protected] | Signature is customized by Netease Mail Master 在2020年03月07日 19:04，Inomjon 写道： 想请教一下，我运行create_traindata生成了5000张图片，batch_size是256训练了20轮，但是在运行test_nn.py文件输出的结果却是完全错误的，想请教一下大家有遇到这种情况吗 Hi brother, did you solve the problem? I encountered same issue — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.
thanks for your reply, do you mean training process? I generated 10000 images, what should I set for epoch? How many epochs should I run?
— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.
Yeah,train the model as long as possible.You can try to run the code before you sleep,and check the performance when you get up. | | buptwanghao | | 邮箱：[email protected] | Signature is customized by Netease Mail Master 在2020年03月09日 15:15，Inomjon 写道： Yeah,because the time you run is too short. I run the code for a whole night,the recognized result was fine. | | buptwanghao | | 邮箱：[email protected] | Signature is customized by Netease Mail Master 在2020年03月07日 19:04，Inomjon 写道： 想请教一下，我运行create_traindata生成了5000张图片，batch_size是256训练了20轮，但是在运行test_nn.py文件输出的结果却是完全错误的，想请教一下大家有遇到这种情况吗 Hi brother, did you solve the problem? I encountered same issue — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe. thanks for your reply, do you mean training process? I generated 10000 images, what should I set for epoch? How many epochs should I run? — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.
thanks!!!
Yeah,train the model as long as possible.You can try to run the code before you sleep,and check the performance when you get up. | | buptwanghao | | 邮箱：[email protected] | Signature is customized by Netease Mail Master 在2020年03月09日 15:15，Inomjon 写道： Yeah,because the time you run is too short. I run the code for a whole night,the recognized result was fine. | | buptwanghao | | 邮箱：[email protected] | Signature is customized by Netease Mail Master 在2020年03月07日 19:04，Inomjon 写道： 想请教一下，我运行create_traindata生成了5000张图片，batch_size是256训练了20轮，但是在运行test_nn.py文件输出的结果却是完全错误的，想请教一下大家有遇到这种情况吗 Hi brother, did you solve the problem? I encountered same issue — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe. thanks for your reply, do you mean training process? I generated 10000 images, what should I set for epoch? How many epochs should I run? — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.
thanks!!!
did you solve the problem by adding the training step? I have encounter the same problem. although i did the same way as above.
