import cv2 as cv
import os
from DataHolder import DataHolder
from PIL import Image
from Classificator import Classificator
from RectangleDetector import RectangleDetector
source_images_folder = "resources\\images\\source_images"
files = [
os.path.join(source_images_folder, p)
for p in sorted(os.listdir(source_images_folder))
]
result = {}
#kls = Classificator.from_folder()
kls = Classificator.from_pickle()
for file in files:
dict_file_names = file.split('\\')
file_name = dict_file_names[len(dict_file_names) - 1]
img = cv.imread(file)
rd = RectangleDetector(img)
cropped_images = rd.crop()
if cropped_images:
entities = []
i = 0
for cropped_image in cropped_images:
i += 1
loc = '{}\\{}-{}'.format("resources\\images\\cropped_images", i,
file_name)
cv.imwrite(loc, cropped_image)
dh = DataHolder(file_name)
