56156540d33f329ebf271ac9313ca723e649c0a2,kraken/blla.py,,segment,#Any#Any#Any#Any#,44

Before Change

    im = im.rotate(angle, expand=True)

    a = pil2array(im)
    binary = np.array(a > 0.5*(np.amin(a) + np.amax(a)), "i")
    binary = 1 - binary

    if mask:
        if mask.mode != "1" and not is_bitonal(mask):
            logger.error("Mask is not bitonal")

After Change

    transforms = dataset.generate_input_transforms(batch, height, width, channels, 0, valid_norm=False)

    with torch.no_grad():
        o = model.nn(transforms(im).unsqueeze(0))
    o = F.interpolate(o, size=im.size[::-1])
    o = segmentation.denoising_hysteresis_thresh(o.detach().squeeze().cpu().numpy(), 0.4, 0.5, 0)
    baselines = segmentation.vectorize_lines(o)
    polygons = segmentation.calculate_polygonal_environment(im, baselines)
    return polygons

Instances