Пример #1
0
Created on Sun Mar 22 20:11:01 2020

@author: Poorvi Rai
"""

from Config import TEST_IMG
from Detector import Detector

from matplotlib import image
from time import time

start_time = time()

img = image.imread(TEST_IMG)

if len(img.shape) == 3:
    imgSingleChannel = img[:, :, 1]
else:
    imgSingleChannel = img

det = Detector()

rectangles = det.scanImgOverScale(imgSingleChannel)

end_time = time()

print("Number of rectangles: ", len(rectangles))
print("Cost time: ", end_time - start_time)

det.showResult(img, rectangles)