运行winpy/main.py遇到的问题

[AzureSkyHuHu]

运行winpy/main.py遇到的问题

[AzureSkyHuHu]

RuntimeError: stack expects each tensor to be equal size, but got [4] at entry 0 and [5] at entry 7

[ypwhs]

你可以检查一下文件夹里的图片，是否都是RGB图片，尺寸是否一致。

如果不一致，你需要手动写一个 collate 函数，对尺寸不一致的图片进行一个补全操作（padding）。
比如先创建一个 (batch_size, height, max_width, 3) 的矩阵，然后将每张图贴进去。

参考链接：

• https://pytorch.org/docs/stable/data.html#working-with-collate-fn

[AzureSkyHuHu]

已解决 谢谢大佬

[Alexuebo]

大佬您好，为什么我检查了我图片尺寸都是一样的（3x70x200)，还是出现了这种问题呢？以及collate_fn函数具体要怎么操作啊

[jjw12345]

我是训练效果很差。。。