关于最后预测图像的问题
Opened this issue · 0 comments
xubigboy commented
image, gt, name = test_loader.load_data()
gt = np.asarray(gt, np.float32)
gt /= (gt.max() + 1e-8)
image = image.cuda()
res5, res4, res3, res2 = model(image)
res = res2
res = F.upsample(res, size=gt.shape, mode='bilinear', align_corners=False)
res = res.sigmoid().data.cpu().numpy().squeeze()
res = (res - res.min()) / (res.max() - res.min() + 1e-8)
misc.imsave(save_path+name, res)
最后保存的图像是一个0-255的图像,而不是黑白二值图,请问是如何到二值化的呢。