refactoring

This commit is contained in:
frankknoll
2023-02-27 12:38:07 +01:00
parent 67579ab7a5
commit 4c311ea8a1

View File

@@ -63,18 +63,12 @@ def _getModel():
return model
def load_model():
_model = keras.models.load_model('model')
_model.make_predict_function()
__model = keras.models.Model(
_model.get_layer(name="image").input,
_model.get_layer(name="dense2").output)
__model.make_predict_function()
return __model
model = keras.models.load_model('model')
return keras.models.Model(
model.get_layer(name="image").input,
model.get_layer(name="dense2").output)
def getTextInCaptchaImage(captchaImageFile):
batchImages = encode_single_sample(captchaImageFile)
model = _getModel()
print('preds = model.predict(batchImages):')
preds = model.predict(batchImages, use_multiprocessing=True)
print('return decode_batch_predictions(preds)[0]:')
preds = _getModel().predict(batchImages, use_multiprocessing=True)
return decode_batch_predictions(preds)[0]