Exemple #1
0
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