【pytorch】【已解决】RuntimeError: stack expects each tensor to be equal size, but got [4, 448, 448] at ent

最新推荐文章于 2025-04-14 14:39:47 发布

「已注销」

最新推荐文章于 2025-04-14 14:39:47 发布

阅读量1.9w

点赞数 2

CC 4.0 BY-SA版权

分类专栏：错误集锦

本文链接：https://blog.youkuaiyun.com/YJYS_ZHX/article/details/114370418

错误集锦专栏收录该内容

7 篇文章

订阅专栏

在使用PyTorch时遇到RuntimeError，原因是数据加载器返回的批次图片尺寸不一致，部分图片为四通道PNG格式。解决方案包括使用skimage或PIL将四通道PNG转换为三通道RGB。在创建数据集时，需确保处理不同通道数的图片。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

详细错误：

Traceback (most recent call last)：
  File "/home/yjys/anaconda3/lib/python3.8/site-packages/torch/utils/data/_utils/collate.py", line 55, in default_collate
    return torch.stack(batch, 0, out=out)
RuntimeError: stack expects each tensor to be equal size, but got [4, 448, 448] at entry 0 and [3, 448, 448] at entry 1

此错误是在创建dataloader后，进行访问时出现，因为trainloader得到的是批量图片，格式为（batch，channel，height，width）。
每次通过__getitem_ _获取单个图片，后经过stack后变为上述的格式。
通过打印每个图片的shape，发现有的shape为（x，x，4），四通道的rgba图片。

skimage解决：
在这里插入图片描述
PIL解决：
PIL将png的RGBA四通道改为jpg的RGB三通道方法