How do I load custom image based datasets into Pytorch for use with a CNN?
Clash Royale CLAN TAG #URR8PPP How do I load custom image based datasets into Pytorch for use with a CNN? I have searched for hours on the internet to find a good solution to my issue. Here is some relevant background information to help you answer my question. This is my first ever deep learning project and I have no idea what I am doing. I know the theory but not the practical elements. The data that I am using can be found on kaggle at this link: (https://www.kaggle.com/alxmamaev/flowers-recognition) I am aiming to classify flowers based on the images provided in the dataset using a CNN. Here is some sample code I have tried to use to load data in so far, this is my best attempt but as I mentioned I am clueless and Pytorch docs didn't offer much help that I could understand at my level. (https://pastebin.com/fNLVW1UW) # Loads the images for use with the CNN. def load_images(image_size=32, batch_size=64, root="../images"): transform = transforms.Compose([ ...