# c++ version pse based on opencv 3+
from pse import decode as pse_decode
from cfglib.config import config as cfg


class TextDetector(object):

    def __init__(self, model):
        # evaluation mode
        self.model = model
        model.eval()
        # parameter
        self.scale = cfg.scale
        self.threshold = cfg.threshold

    def detect(self, image, img_show):
        # get model output
        preds = self.model.forward(image)
        preds, boxes, contours = pse_decode(preds[0], self.scale, self.threshold)

        output = {
            'image': image,
            'tr': preds,
            'bbox': boxes
        }
        return contours, output