search

bigmb / Unet-Segmentation-Pytorch-Nest-of-Unets

Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet
MIT License
1.87k stars 345 forks source link

dim #55

Closed giter915 closed 2 years ago

giter915 commented 2 years ago

File "F:\some_project\Unet-Segmentation-Pytorch-Nest-of-Unets-master\Unet-Segmentation-Pytorch-Nest-of-Unets-master\Models.py", line 102, in forward d5 = torch.cat((e4, d5), dim=1) RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 61 but got size 60 for tensor number 1 in the list.

bigmb commented 2 years ago

Whats the data shape? Make sure its in muliple of 2^4. Or reduce the depth of the model. (You can check other closed issue for this too.)