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

Before Change

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

    o = nn.nn(im)
    o = segmentation.denoising_hysteresis_thresh(o, 0.4, 0.5, 0)
    baselines = segmentation.vectorize_lines(o)

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