Initial commit
- find receipt - find lines
This commit is contained in:
20
main.py
Normal file
20
main.py
Normal file
@@ -0,0 +1,20 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user