How to build custom Datasets for Images in Pytorch

In this video we have downloaded images online and store them in a folder together with a csv file and we want to load them efficiently with a custom Dataset in Pytorch.

Small Example of Dataset used in video:

Dataset (All images but not with csv file so you have to create it youself):

❤️ Support the channel ❤️

Paid Courses I recommend for learning (affiliate links, no extra cost for you):

✨ Free Resources that are great:

💻 My Deep Learning Setup and Recording Setup:

GitHub Repository:

✅ One-Time Donations:

▶️ You Can Connect with me on:

[kiranraghavendra]

Your series of pytorch tutorials has been legit amazing. Keep up the good work!!

[colorful_face]

Finally! Was looking how to do this for a while now.. was wondering how to load text files from a directory, but now I can adopt the code and do it!

[widdalightsout]

Thankyou so much, I was stuck on this custom dataset problem for a while now. Really appreciate it.

[rbaleksandar]

I recommend using:

train_size = int(X * len(dataset))
test_size = len(dataset) - train_size
train_set, test_set = torch.utils.data.random_split(dataset, [train_size, test_size])

where X is a percentage as float e.g. 0.6 for 60% of the dataset to be used for training. Since I'm quite new to ML, PyTorch etc. it took me some time to figure out how to split the set (I used the small dataset of 10 images). I was getting an error regarding the "Sum of input lengths does not equal the length of the input dataset!".


I couldn't find any use of in_channel, num_classes, batch_size etc in your code.


Thanks for the tutorial! It really helped me a lot!

[peterchen]

GOD THANK YOU SO MUCH I've been looking for this for SO LONG you're a survior

[DharmikBhatt]

This video helped me understand how to make any kind of custom dataset! thank you

[hilta]

great tutorial, i find it's easy to understand! Thanks!

[anko]

So thanks. I've been looking for this !

[luciachen]

Thank you so much! It's so helpful for my course project🥰

[aajanquail]

Thank you so much! This was enormously helpful.

[copuzvv]

Thank you for tutorial! This was very helpful!

[unacan]

This is the most clear to understand ever! Many thanks👍

[ripleylee]

Thank you for this video! It really helps!

[mrtenaviv]

Thanks so much for your guide. Finally I found how to make my own dataset

[keshavbalachandar]

This was perfect, thank you so so much :)

[asiasowa]

What a great video! Thanks! it's helped me a lot! :)

[seolakim]

Thank you enormously for the video, you are the best!!

[ashfakyeafi]

this is amazing tutorial. thanks for making this 👍👍👍

[more]

Maybe the only exalmpe of how to ACTUALLY use the frameworks. It all starts with importing our own data to the model and how to do it. PyTorch and TensorFlow are quite easy to be used just by following the instructions anyway. Thank you so much!

[dulampradeep]

thanks sir its very helpful to the beginners