Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)
2.8k
stars
581
forks
source link
Why Minus 1 ? #153
Open
Kittywyk opened 3 years ago
def _mask_transform(self, mask): return torch.LongTensor(np.array(mask).astype('int32') - 1)
Thx for ur explanation.