qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-04-02 05:21:58 +02:00
Code Issues Projects Releases Wiki Activity

BUG fixing for cnn-rnn ocr model training if augmentation is false

Browse source
This commit is contained in:
vahidrezanezhad 2026-03-27 09:15:19 +01:00
parent 7499e3e7b8
commit 8333158ecc
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/eynollah/training/utils.py
View file

@ -1570,7 +1570,7 @@ def data_gen_ocr(
if to_yield: yield to_yield
else:
img_out, batchcount, ret_x, ret_y, to_yield = increment_batchcount(img_out, batchcount, ret_x, ret_y)
img_out, batchcount, ret_x, ret_y, to_yield = increment_batchcount(img, batchcount, ret_x, ret_y)
if to_yield: yield to_yield