Update e2emodel.py

This commit is contained in:
Jack Yu
2017-11-17 18:50:52 +08:00
committed by GitHub
parent 4ce9e3b195
commit 70fd1621a6
+1 -2
View File
@@ -2,7 +2,6 @@
from keras import backend as K from keras import backend as K
from keras.models import * from keras.models import *
from keras.layers import * from keras.layers import *
from captcha.image import ImageCaptcha
import e2e import e2e
@@ -32,4 +31,4 @@ def construct_model(model_path):
x = Activation('softmax')(x) x = Activation('softmax')(x)
base_model = Model(inputs=input_tensor, outputs=x) base_model = Model(inputs=input_tensor, outputs=x)
base_model.load_weights(model_path) base_model.load_weights(model_path)
return base_model return base_model