faustomorales / keras-ocr

A packaged and flexible version of the CRAFT text detector and Keras CRNN recognition model.
https://keras-ocr.readthedocs.io/
MIT License
1.36k stars 347 forks source link

Error when training detector with batch size > 1 #218

Open nguyenquyem99dt opened 1 year ago

nguyenquyem99dt commented 1 year ago

The function get_batch_generator just work correctly if we use batch_size=1. I think the module detection.py L:727 must be X = np.array([compute_input(image) for image in images], dtype="float32") instead of X = compute_input(images) if we want to use batch_size > 1.