mirror of
https://github.com/qurator-spk/eynollah.git
synced 2026-04-02 13:31:59 +02:00
BUG fixing for cnn-rnn ocr model training if augmentation is false
This commit is contained in:
parent
7499e3e7b8
commit
8333158ecc
1 changed files with 1 additions and 1 deletions
|
|
@ -1570,7 +1570,7 @@ def data_gen_ocr(
|
||||||
if to_yield: yield to_yield
|
if to_yield: yield to_yield
|
||||||
|
|
||||||
else:
|
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
|
if to_yield: yield to_yield
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue