def crop(self):
# image path and valid extensions
# imageDir = "~/RomanNumberDetector/numbers" # specify your path here
# imageDir = os.path.abspath("C:/Code/PREN2/RomanNumberDetector/numbers")
imageDir = os.path.join(os.path.dirname(os.path.realpath(__file__)),
"numbers")
print(imageDir)
image_path_list = []
valid_image_extensions = [".tiff"] # specify your vald extensions here
valid_image_extensions = [
item.lower() for item in valid_image_extensions
]
# create a list all files in directory and
# append files with a vaild extention to image_path_list
for file in os.listdir(imageDir):
extension = os.path.splitext(file)[1]
if extension.lower() not in valid_image_extensions:
continue
image_path_list.append(os.path.join(imageDir, file))
# loop through image_path_list to open each image
for imagePath in image_path_list:
img = cv2.imread(imagePath)
colorFilter = crop.ColorFilter(img)
colorFilter.filterRed()
colorFilter.closing(colorFilter.mask)
# colorFilter.showImg('closing',colorFilter.closing)
shapeD = crop.ShapeDetecter(img, colorFilter.closing)
self.cropped.append(shapeD.analyse())
def crop(self):
self.cropped.clear()
for image in self.detectedRomanBars:
img = image
colorFilter = crop.ColorFilter(img)
colorFilter.filterRed()
colorFilter.closing(colorFilter.mask)
# colorFilter.showImg('closing',colorFilter.closing)
shapeD = crop.ShapeDetecter(img, colorFilter.closing)
self.cropped.append(shapeD.analyse())
return self.analyse()
# append files with a vaild extention to image_path_list
for file in os.listdir(imageDir):
extension = os.path.splitext(file)[1]
if extension.lower() not in valid_image_extensions:
continue
image_path_list.append(os.path.join(imageDir, file))
# loop through image_path_list to open each image
for imagePath in image_path_list:
img = cv2.imread(imagePath)
colorFilter = crop.ColorFilter(img)
colorFilter.filterRed()
colorFilter.closing(colorFilter.mask)
# colorFilter.showImg('closing',colorFilter.closing)
shapeD = crop.ShapeDetecter(img, colorFilter.closing)
cropped.append(shapeD.analyse())
############################################
# analyse Pictures
############################################
detectedNumber = []
for abc in cropped:
detectedNumber.append(analyse.analyseNumber(abc))
print(detectedNumber)
count = [0] * 5