Hi! I am trying to make custom dataset for HSI (HxWxD) and groundtruth (HxW) data stored in .mat file.
Which i would like to use in:
train_loader = torch.utils.data.DataLoader()
I found class like:
from torch.utils.data.dataset import Dataset
class MyCustomDataset(Dataset):
def init(self, …):
# stuff
def getitem(self, index):
# stuff
return (img, label)
def len(self):
return count
To make further 3D CNN mdel for classification.
Could you give me some help or examples for this problem of mine?
Or is this even necessary for making custom data class for cnn?
Thank you in advance!