如何解决有没有办法通过 Pytorch 将一组数据作为一个整体进行分类?
我目前正在处理 CT 数据集上的分类任务。在 CT 数据集中,多个切片属于一个患者,在设置我的数据集时,我将数据排列如下: 数据集/0/patient_1/1.png,2.png... 数据集/0/patient_2/1.png,2.png... 我想知道有没有办法让我的网络按患者而不是切片进行分类?
谢谢
解决方法
每个切片都是 2D 图像,而对于每个患者,您都有一个 3D 体积的 CT 体素。
如果您想针对每个患者而不是每个切片进行工作,则需要组织数据以输出成批的 3D 信息(形状为 batch
xchannel
xdepth
x height
xwidth
) 并使您的模型处理 3D 信息(例如,使用 Conv3D
而不是 Conv2D
)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。