Thank you for opening my new conversation.I have little chances to speak English ,so my English is bad and maybe the way that I express is not appropriate.I am sorry for this. I put some pictures in folder and I use the ImageFolder to get my datasat.What i would like to know is if the dataset could bu used in net without changing to torch.utils.data.TensorDataset ?
If you have a folder structure similar to the imagenet one (each folder represents a class, and has images in it), then you can directly use the
torchvision.dataset.ImageFolder, and no need to go through the
torch.utils.data.TensorDataset. Here is an example.