【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完全相同;
参考
完
各美其美,美美与共,不和他人作比较,不对他人有期待,不批判他人,不钻牛角尖。
心正意诚,做自己该做的事情,做自己喜欢做的事情,安静做一枚有思想的技术媛。
版权声明,转载请注明出处:https://www.cnblogs.com/happyamyhope/
心正意诚,做自己该做的事情,做自己喜欢做的事情,安静做一枚有思想的技术媛。
版权声明,转载请注明出处:https://www.cnblogs.com/happyamyhope/