python - pytorch 中轴与数组不匹配错误

Question

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

标准化扫描后

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

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

axes don't match array error.

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

Answer 1

注释第 417 行和 418 行。问题将得到解决

问题是因为这两行

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