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.