轴不匹配pytorch中的数组错误

Question

我试图在pytorch中运行vnet实现 （https://github.com/mattmacy/vnet.pytorch）并在使用

规范化扫描之后

x_max = 512 
y_max = 512 
z_max = 500 
voxspacing = 0.7 

当我在for循环通过数据加载器枚举的行上调用tran函数时，我得到了一个

  

轴不匹配数组错误。

我不明白如何解决这个问题。任何帮助都会非常有帮助。我是机器学习的新手

Answer 1

对第417和418行进行评论。问题将得到解决

问题是因为这两行

if self.transform is not None:
     img = self.transform(img)