pytorch 没有按预期读取输入通道

Question

我正在使用 pytorch 为视频构建 conv3d 模型。输入是

(2, 30, 46, 140, 1)

但 pytorch 读取输入通道是第二个，而实际上是第四个。

        self.conv1 = nn.Conv3d(in_channels=1, out_channels=32, kernel_size=3, padding=1)
Given groups=1, weight of size [32, 1, 3, 3, 3], expected input[2, 30, 46, 140, 1] to have 1 channels, but got 30 channels instead

我尝试将输入重塑为

(2, 1, 30, 46, 140)

，但随后它不显示框架并给出输入错误的错误

TypeError: Invalid shape (30, 46, 140) for image data

请记住，我尝试在张量流上使用相同的输入运行相同的模型，并且它有效，但由于依赖性问题，我无法使用张量流

Answer 1

我解决了问题，我在自定义数据集类中转置了形状，然后重新转置它以查看帧图像