pytorch - 有没有办法通过 Pytorch 对一组数据进行整体分类？

Question

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

谢谢你

标签： pytorchclassification

Answer 1

每个切片都是 2D 图像，而对于每个患者，您都有一个 3D 体积的 CT 体素。
如果您想按患者而不是按切片工作，则需要组织数据以输出批量 3D 信息（形状为batchx channelx depthx heightx width）并使模型处理 3D 信息（例如，使用Conv3D而不是Conv2D)