Files
receipt-recognition/main.py
Caesar2011 8626a2db01 Initial commit
- find receipt
 - find lines
2019-03-16 17:06:40 +01:00

21 lines
643 B
Python

import glob
import os
from src.processing.loader import load_image, save_image
from src.processing.linefinder import find_lines, preparation
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"):
image = load_image("data/"+file)
receipt = cut_receipt(image, draw_steps=True)
plt.imshow(receipt)
plt.show()
lines = find_lines(receipt)
for line in lines:
plt.imshow(line, cmap="gray")
plt.show()