CRAFT Prediction and post-processing

predict_craft.py

# Load image as torch tensor
x = torch.from_numpy(x).permute(2, 0, 1)    # [h, w, c] to [c, h, w]
x = Variable(x.unsqueeze(0))                # [c, h, w] to [b, c, h, w]

if cuda:
    x = x.cuda()

# forward pass
with torch.no_grad():
    y, feature = net(x)

# Get Score and Affinity Map
score_text = y[0,:,:,0].cpu().data.numpy()
score_link = y[0,:,:,1].cpu().data.numpy()