francois/HowBadIsMyBatch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
frankknoll
2023-02-27 12:38:07 +01:00
parent 67579ab7a5
commit 4c311ea8a1
1 changed files with 5 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/CaptchaReader.py
View File

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