from app.dataset import Dataset from app.output import Output from app.segmenter import Segmenter from app.utils import Utils from app.debug import Debug import numpy as np import cv2 dataset = Dataset() dataset.load() segmenter = Segmenter() Output.clear() for item in dataset.items: print(item.file) #if item.file != "008.jpg": # continue # # run the segmentation algorithm # quad = segmenter.segment(item.img, item.distribution_rect) # # draw the result # # reference image