def _segment(self, doc, sort_by_columns, expected):
self.assertEqual(5, len(doc.items))
doc.set_items([])
self.assertEqual(0, len(doc.items))
segment_doc = SegmentDocument(sort_by_columns=sort_by_columns)
doc, display_image = segment_doc.segment(doc)
# Compare the rects in pixels
actual = doc.scanned.from_normalised([i['rect'] for i in doc.items])
self.assertEqual(list(expected), list(actual))
def _segment(self, doc, sort_by_columns, expected):
self.assertEqual(5, len(doc.items))
doc.set_items([])
self.assertEqual(0, len(doc.items))
segment_doc = SegmentDocument(sort_by_columns=sort_by_columns)
doc, display_image = segment_doc.segment(doc)
# Compare the rects in pixels
actual = doc.scanned.from_normalised([i['rect'] for i in doc.items])
self.assertEqual(list(expected), list(actual))
def segment(dir, sort_by_columns):
dir = Path(dir)
segment_doc = SegmentDocument(sort_by_columns)
for p in dir.glob('*' + InselectDocument.EXTENSION):
doc = InselectDocument.load(p)
if not doc.items:
print('Segmenting [{0}]'.format(p))
try:
debug_print('Will segment [{0}]'.format(p))
doc, display_image = segment_doc.segment(doc)
del display_image # We don't use this
doc.save()
except KeyboardInterrupt:
raise
except Exception:
print('Error segmenting [{0}]'.format(p))
traceback.print_exc()
else:
print('Segmented [{0}]'.format(doc))
else:
print('Skipping [{0}] as it already contains items'.format(p))
def segment(dir, sort_by_columns):
dir = Path(dir)
segment_doc = SegmentDocument(sort_by_columns)
for p in dir.glob('*' + InselectDocument.EXTENSION):
doc = InselectDocument.load(p)
if not doc.items:
print('Segmenting [{0}]'.format(p))
try:
debug_print('Will segment [{0}]'.format(p))
doc, display_image = segment_doc.segment(doc)
del display_image # We don't use this
doc.save()
except KeyboardInterrupt:
raise
except Exception:
print('Error segmenting [{0}]'.format(p))
traceback.print_exc()
else:
print('Segmented [{0}]'.format(doc))
else:
print('Skipping [{0}] as it already contains items'.format(p))
