Conv3D模型输入张量

Question

我是PyTorch的新手，我想为3D DICOM MRI制作一个分类器。我想使用来自monai库的预先训练过的resnet18，但是我对张量的输入维度感到困惑。在我的数据采集器中，图像的形状是2160,256,256，其中2是batch_size，160是每个病人的dicom图像数，256 x256是图像的维数。当我试图运行模型时，我得到了这样的错误：期望5维权重为64，3，7，7，7，7，7，7，7，7，7，7，7，7，7，7维输入，但是得到了2,160,256,256的4维输入，而不是。

如果我解开张量，然后再把它给我得到的模型：给定groups=1，尺寸64，3，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7，7

Answer 1

您需要为每个切片添加通道维度(这是一个用于MRI的)。你的输入应该是形状的(2,1,160,256,256)