Update e2emodel.py
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user