pspnet-pytorch
pspnet-pytorch copied to clipboard
Size mismatch
When initializing the model, it gives size mismatch error. Input image size: 3, 224, 224 Output Image size: 1, 224, 224 n_classes=1 What could be the issue?