pytorch-semantic-segmentation icon indicating copy to clipboard operation
pytorch-semantic-segmentation copied to clipboard

Published 20 hours ago • verified publisher icon zijundeng

Assertion errors

Open harishkrishnav opened this issue 8 years ago • 3 comments
trafficstars

I am facing size mismatch issues while training cityscapes. Primarily, the labels are of shape (16L, 512L, 256L, 3L) while the outputs are of shape (16L, 19L, 512L, 256L).

harishkrishnav avatar Oct 10 '17 01:10 harishkrishnav

Also, how was this resolved?

harishkrishnav avatar Oct 10 '17 01:10 harishkrishnav

The label batch should be in shape (16, 512, 256) and the output batch should be in shape (16, 19, 512, 256). Check whether you loaded the wrong label image.

zijundeng avatar Oct 11 '17 15:10 zijundeng

For this issue you should check whether torchvision.transforms.ToTensor is used for the label image.

zijundeng avatar Oct 11 '17 15:10 zijundeng