【pytorch基础-error】RuntimeError: stack expects each tensor to be equal size

 

问题

File "/home/xxx/miniconda3/envs/open_mmlab/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 [3, 867, 867] at entry 0 and [3, 740, 740] at entry 1

 

原因：

最后发现DataLoader输入的数据集的大小必须是一致的，如果不一致需要使用collate_fn选项处理成一致的；

最后发现还是增加的随机裁剪的过程改变了数据的大小，不能保证最后是一致的；自以为某些地方没有问题，没有深入去理解实现过程以及最后的结果的性质，单个函数可能理解了，但是一系列串联的数据操作(RandomResize、RandomCrop等)没有联系到一起，虽然单个操作可能会改变大小，但是串联起来之后最后的结果肯定是一致的；

 

也就是，中间的处理过程可以是不一致的，但是最后进入DataLoader的数据必须是一致的；

这是因为torch.stack的实现过程决定的；因为stack要求两个输入的shape完全相同；

 

参考

1. torch.tensor拼接与list(tensors)；

2. pytorch长度不同的数据如何放在一个batch；

3. 使用Pytorch的dataloader时报错每个tensor的维度不一样；

完