From 4c311ea8a18792fb91c1dc0c3022916f09702b3e Mon Sep 17 00:00:00 2001 From: frankknoll Date: Mon, 27 Feb 2023 12:38:07 +0100 Subject: [PATCH] refactoring --- src/CaptchaReader.py | 16 +++++----------- 1 file changed, 5 insertions(+), 11 deletions(-) diff --git a/src/CaptchaReader.py b/src/CaptchaReader.py index f7902561504..5d2461e7b78 100644 --- a/src/CaptchaReader.py +++ b/src/CaptchaReader.py @@ -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]