Question

这是线性自动编码器代码，原始图片为344 * 344 RGB，训练过程结束后，我想使用下面的代码显示解码后的图片，但是它具有ValueError：轴不匹配数组

pytorch，google colab（GPU）

  const handleSubmit = () => {

    /* 
    Update the list state to a new array, with newItem appended to the
    end of it. This ensures the list state remains as an array type,
    ensuring the list.map() is defined 
    */
    setlist([...list, newItem]);
  };

我希望输出2张图片，但现在只显示原始图片，而解码后的图片不显示。

训练过程效果很好，但是我不知道图片的尺寸有什么问题。

Answer 1

decoder返回形状为BATCH_SIZE x 355008的线性输出。首先，我们需要在将第二维变形为形状3 x 344 x 344的3维之前，在其上应用转置。用以下内容替换decodeddd应该可以解决问题：

decodeddd = np.transpose(decoded.cpu()[0].view(3, 344, 344),(1,2,0))

轴与阵列不匹配/尺寸不匹配，m1：[132096 x 344]，m2：[118336 x 128]

1 个答案: