Skip to content
Snippets Groups Projects
Commit 091ecb88 authored by Nikos Athanasiou's avatar Nikos Athanasiou
Browse files

bug in sample_seq

parent 8dd7a47e
No related branches found
No related tags found
No related merge requests found
......@@ -161,8 +161,8 @@ def render_cli(cfg: DictConfig) -> None:
with Image(filename = p) as img:
draw.font = 'Helvetica-Bold'
draw.font_size = 30
draw.text_under_color = Color('gray30')
draw.fill_color = Color('white')
# draw.text_under_color = Color('gray30')
# draw.fill_color = Color('white')
draw.text_alignment = 'center'
draw.text(int(img.width / 2)-5, img.height - 15, f'{text[i]}')
draw(img)
......
......@@ -171,11 +171,10 @@ def sample(newcfg: DictConfig) -> None:
one_data = dataset.load_keyid(keyid)
# buggy gt
if dataset_val.dtype == 'separate_pairs':
if one_data['length_0'] == 1 or one_data['length_1'] == 1 :
logger.info(f'Omitted {keyid}')
ommited += 1
continue
if one_data['length_0'] == 1 or one_data['length_1'] == 1 :
logger.info(f'Omitted {keyid}')
ommited += 1
continue
batch = collate_pairs_and_text([one_data])
cur_lens = [batch['length_0'][0], batch['length_1'][0] + batch['length_transition'][0]]
cur_texts = [batch['text_0'][0], batch['text_1'][0]]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment