有没有办法通过Pytorch将一组数据作为一个整体进行分类？

我目前正在处理CT数据集的分类任务。在CT数据集中，多个切片属于一个患者，在设置数据集时，我将数据安排如下： 数据集/0/patient_1/1.png，2.png。。。 数据集/0/patient_2/1.png，2.png。。。 我想知道是否有一种方法可以让我的人际网络按患者而不是按切片进行分类

---

谢谢

每个切片都是2D图像，而每个患者都有3D体积的CT体素。

---

如果您想针对每个患者而不是每个切片工作，则需要组织数据以输出成批的3D信息（形状

batch

x

channel

x

depth

x

height

x

width

），并使模型处理3D信息（例如，使用

Conv3D

而不是

Conv2D

）

谢谢你的宝贵建议！考虑到我有限的计算资源，我可能很难使用3D网络，所以我正在寻找2D解决方案。只要你能在GPU中安装一个卷，你就可以