From d202ed1b425af6ae86f48a21724c193d6930c096 Mon Sep 17 00:00:00 2001
From: Martin <maarand@teklia.com>
Date: Tue, 5 Nov 2019 15:51:52 +0100
Subject: [PATCH] fix line_id bug
---
kaldi_data_generator.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/kaldi_data_generator.py b/kaldi_data_generator.py
index 86f50c0..b955dbd 100644
--- a/kaldi_data_generator.py
+++ b/kaldi_data_generator.py
@@ -94,10 +94,10 @@ class KaldiDataGenerator:
for i, [x, y, w, h] in enumerate(line_bounding_rects):
cropped = img[y:y + h, x:x + w].copy()
- cv2.imwrite(f'{self.out_line_img_dir}_{i}.jpg', cropped)
+ cv2.imwrite(f'{self.out_line_img_dir}/{page_id}_{i}.jpg', cropped)
for i, text in enumerate(line_transcriptions):
- write_file(f"{self.out_line_text_dir}_{i}.txt", text)
+ write_file(f"{self.out_line_text_dir}/{page_id}_{i}.txt", text)
def run_pages(self, page_ids):
for page_id in page_ids:
--
GitLab