qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-02-20 16:32:03 +01:00
Code Issues Projects Releases Wiki Activity

fix typo in 27f43c17

Browse source
This commit is contained in:
Robert Sachunsky 2026-02-17 17:48:43 +01:00
parent c1b5cc92af
commit 6a4163ae56
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -523,7 +523,7 @@ def run(_config,
) )
train_ds = tf.data.Dataset.from_generator(gen) train_ds = tf.data.Dataset.from_generator(gen)
train_ds = train_ds.padded_batch(n_batch, train_ds = train_ds.padded_batch(n_batch,
padded_shapes=([image_height, image_width, 3], [None]), padded_shapes=([input_height, input_width, 3], [None]),
padding_values=(0, padding_token), padding_values=(0, padding_token),
drop_remainder=True, drop_remainder=True,
#num_parallel_calls=tf.data.AUTOTUNE, #num_parallel_calls=tf.data.AUTOTUNE,

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

@ -997,12 +997,12 @@ def preprocess_img(img,
input_height, input_height,
input_width) input_width)
if padding_black: if padding_black:
yield from get_patches(do_padding_black(img), yield from get_patches(do_padding_with_color(img, 'black'),
do_padding_label(lab), do_padding_label(lab),
input_height, input_height,
input_width) input_width)
if padding_white: if padding_white:
yield from get_patches(do_padding_white(img), yield from get_patches(do_padding_with_color(img, 'white'),
do_padding_label(lab), do_padding_label(lab),
input_height, input_height,
input_width) input_width)
@ -1129,7 +1129,7 @@ def preprocess_img_ocr(
return scale_padd_image_for_ocr(img, input_height, input_width).astype(np.float32) / 255. return scale_padd_image_for_ocr(img, input_height, input_width).astype(np.float32) / 255.
#lab = vectorize_label(lab, char_to_num, padding_token, max_len) #lab = vectorize_label(lab, char_to_num, padding_token, max_len)
# now padded at Dataset.padded_batch # now padded at Dataset.padded_batch
lab = char_to_num(tf.strings.unicode_split(label, input_encoding="UTF-8")) lab = char_to_num(tf.strings.unicode_split(lab, input_encoding="UTF-8"))
yield scale_image(img), lab yield scale_image(img), lab
#to_yield = {"image": ret_x, "label": ret_y} #to_yield = {"image": ret_x, "label": ret_y}