JavisPeng / u_net_liver

MIT License
201 stars 75 forks source link

如何在可视化结果中用绿色标注目标? #8

Closed li-zheng-hao closed 4 years ago

li-zheng-hao commented 4 years ago
    with torch.no_grad():
        for x, _ in dataloaders:
            y=model(x)
            img_y=torch.squeeze(y).numpy()
            plt.imshow(img_y)

请问测试代码中这一段img_y输出的值代表什么呢?如何用绿色或者红色标注出图像中的目标?