Improved line finding
This commit is contained in:
10
main.py
10
main.py
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user