From 8333158eccab7d6e17607d8ebee8c2454f809754 Mon Sep 17 00:00:00 2001 From: vahidrezanezhad Date: Fri, 27 Mar 2026 09:15:19 +0100 Subject: [PATCH] BUG fixing for cnn-rnn ocr model training if augmentation is false --- src/eynollah/training/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/eynollah/training/utils.py b/src/eynollah/training/utils.py index d44f10c..aef67a0 100644 --- a/src/eynollah/training/utils.py +++ b/src/eynollah/training/utils.py @@ -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