Expected size 24 but got size 25 for tensor numbe

举报
AI浩 发表于 2022/04/11 07:39:14 2022/04/11
【摘要】 在做图像分割的时候遇到了错误,错误如下: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

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。