Write results to <img_filename>.txt files.

This commit is contained in:
Andrey Golovizin 2014-09-04 21:44:00 +02:00
parent 76ffaf5ad3
commit a75630890e

View file

@ -55,8 +55,7 @@ class OCREngine(QThread):
return Page(Image.fromfile(filename), filename) return Page(Image.fromfile(filename), filename)
def run(self): def run(self):
for page_text in self.recognize(): self.recognize()
print(page_text)
if self.quit: if self.quit:
qApp.quit() qApp.quit()
@ -64,7 +63,10 @@ class OCREngine(QThread):
for filename in self.filenames: for filename in self.filenames:
page = self.load_page(filename) page = self.load_page(filename)
self.pageChanged.emit(page) self.pageChanged.emit(page)
yield self.recognize_page(page) page_text = self.recognize_page(page)
print(page_text)
with open(filename + '.txt', 'w') as page_text_file:
page_text_file.write(page_text)
def recognize_page(self, page): def recognize_page(self, page):
glyph_data_seq = itertools.chain(*(self.recognize_line(line) for line in page.lines)) glyph_data_seq = itertools.chain(*(self.recognize_line(line) for line in page.lines))