search

prithuls / MV-Swin-T

9 stars 0 forks source link

data structure #2

Closed emrekeles-arch closed 5 months ago

emrekeles-arch commented 5 months ago

How should the data structure be for training ?

prithuls commented 5 months ago

Thanks for your interest in our research.

For torch.utils.data.DataLoader, the parameter is a function that returns image_cc, image_mlo, and label in that order. For image_cc and image_mlo, we used PIL.Image.

Hope this helps.