Expected size 24 but got size 25 for tensor numbe
【摘要】 在做图像分割的时候遇到了错误,错误如下:File "D:/segmentation/Pytorch-UNet-master/train.py", line 193, in <module> amp=args.amp) File "D:/segmentation/Pytorch-UNet-master/train.py", line 88, in train_net masks_...
在做图像分割的时候遇到了错误,错误如下:
File "D:/segmentation/Pytorch-UNet-master/train.py", line 193, in <module>
amp=args.amp)
File "D:/segmentation/Pytorch-UNet-master/train.py", line 88, in train_net
masks_pred = net(images)
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\segmentation_models_pytorch\base\model.py", line 16, in forward
decoder_output = self.decoder(*features)
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\segmentation_models_pytorch\unet\decoder.py", line 121, in forward
x = decoder_block(x, skip)
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "D:\ProgramData\Anaconda3\envs\mytorch\lib\site-packages\segmentation_models_pytorch\unet\decoder.py", line 39, in forward
x = torch.cat([x, skip], dim=1)
RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 24 but got size 25 for tensor number 1 in the list.
从错误的分析得知是维度没有对上。将x和skip的维度打印出来:
torch.Size([2, 448, 24, 18]) torch.Size([2, 160, 25, 18])
这就对应上上面的错误了。
出现这个错误的原因是在输入图片的时候,没有将图像resize成512×512大小,导致维度不一致!
将图片resize后就可以解决。
在打印x和skip的维度:
torch.Size([2, 448, 32, 32]) torch.Size([2, 160, 32, 32])
【声明】本内容来自华为云开发者社区博主,不代表华为云及华为云开发者社区的观点和立场。转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息,否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论(0)