pytorch cross-entropy 用于图像分割(多类别)
1、predict_result and mask#我们假设每个像素点可能属于3个类别之一:0 or 1 or 2#注意mask的shape=(1,2,2),我们要指定batch_size:1,但是不用指定channel,不用对mask做one_hot处理#四个像素:four piexlsmask = torch.tensor([[[1,2],[2,0]]])#网络的输出shape=(1,3,2,2),batch_size=1,channel=3pre = torch.tensor(.
原创
2020-09-06 20:57:50 ·
2614 阅读 ·
0 评论