classification - Is there a way to classify a set of data as a whole via Pytorch?

Question

I'm currently dealing with a classification task on a CT dataset. In CT datasets, multiple slices belong to one single patient, while setting up my dataset, I arrange my data as follows: dataset/0/patient_1/1.png,2.png... dataset/0/patient_2/1.png,2.png... I wonder is there a way to let my network to classify by patient instead of by slices?

thank you

question from:https://stackoverflow.com/questions/65895418/is-there-a-way-to-classify-a-set-of-data-as-a-whole-via-pytorch

Answer 1

Each slice is a 2D image, while for each patient you have a 3D volume of CT voxels.
If you want to work per-patient, rather than per-slice, you'll need to organize your data to output batches of 3D information (of shape batchxchannelxdepthxheightxwidth) and make your model process 3D information (e.g., using Conv3D instead of Conv2D)