Improved line finding

This commit is contained in:
Caesar2011
2019-03-20 22:20:17 +01:00
parent 56d526d00f
commit 516efb0d37
3 changed files with 40 additions and 18 deletions

10
main.py
View File

@@ -1,18 +1,20 @@
import glob
import os
from src.processing.loader import load_image, save_image
from src.processing.linefinder import find_lines, preparation
from src.processing.loader import load_image
from src.processing.linefinder import find_lines
from matplotlib import pyplot as plt
from src.processing.receiptcutter import cut_receipt
for root, dirs, files in os.walk("data"):
for file in files:
if file.startswith("receipt-08"):
if file.startswith("receipt-12"):
image = load_image("data/"+file)
receipt = cut_receipt(image, draw_steps=True)
lines = find_lines(receipt)
lines = find_lines(receipt, draw_steps=True)
print("hh", len(lines))
#continue
for line in lines:
plt.imshow(line, cmap="gray")
plt.show()