【问题标题】:Axes don't match array error in pytorch轴与pytorch中的数组错误不匹配
【发布时间】:2018-01-18 23:52:28
【问题描述】:

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

标准化扫描后
x_max = 512 
y_max = 512 
z_max = 500 
voxspacing = 0.7 

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

轴不匹配数组错误。

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

【问题讨论】:

  • tran 有什么功能?你的意思是train 还是transform?我也找不到您所指的x_maxy_maxz_max 甚至voxspacing
  • 在火车功能中。 x_max、y_max、z_max、voxspacing 位于 torchbiomed lib、数据集中、luna16.py 文件中。它用于使所有文件具有相同的大小和相同的填充

标签: python python-3.x machine-learning conv-neural-network pytorch


【解决方案1】:

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

问题在于这两行

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

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2021-03-11
    • 1970-01-01
    • 1970-01-01
    • 2019-06-16
    • 1970-01-01
    • 2020-09-21
    相关资源
    最近更新 更多