Image Classification Using Pytorch and Convolutional Neural Network

This video provides a comprehensive guide on creating an image classification model using PyTorch and Convolutional Neural Networks (CNNs). We dive into the world of deep learning, focusing on the development of a custom dataset to train and evaluate our model. Whether you're a beginner looking to get started with image classification or an enthusiast seeking to enhance your PyTorch and CNN skills, this video is the perfect resource for you.
Github: github.com/Aar...
For queries: You can comment in comment section or you can mail me at [e-mail]
#imageclassification #computervision #pytorch #cnn #convolutionalneuralnetworks #convolutionalneuralnetwork

[@SagarLekhak]

How can someone explain such complex concepts in a very simple way? I adore you.

[@CodeWithAarohi]

Glad my video is helpful 🙂

[@arnavthakur5409]

Keep sharing such an amazing knowledgeable content in form of very easy to learn videos.

[@CodeWithAarohi]

Thank you, I will

[@Sunil-ez1hx]

Hello Ma’am Your AI and Data Science content is consistently impressive! Thanks for making complex concepts so accessible. Keep up the great work! 🚀 #ArtificialIntelligence #DataScience #ImpressiveContent 👏👍

[@CodeWithAarohi]

My pleasure 😊

[@josephmyalla3611]

Great, Short and Clear

[@CodeWithAarohi]

Thanks!

[@mainhoontom2176]

Very nice Aarohi Mam. Thanks for making complex stuff simple.

[@CodeWithAarohi]

Most welcome 😊

[@soravsingla8782]

Really knowledgeable video & explained in a Very well manner. Thank you

[@CodeWithAarohi]

Glad it was helpful!

[@yasharazadvatan6673]

Hi, thank you for sharing this great content. I have a question; in 19th minute of the video, you create a model and load the trained model. also you create new_model variable. in 20th minute of the video, you write output = model(input_batch) I get confused, where we use new_model?

[@karthickkuduva9819]

thanks for such easy tutorial on image classification mam.... worth watching your channel

[@CodeWithAarohi]

Glad to hear that

[@ashimasingla103]

Hello Aarohi Your channel is very knowledgeable & helpful for all Artificial Intelligence/ Data Scientist Professionals. Stay blessed & keep sharing such a good content. Your channel really needs more likes & share so to reach maximum AI professionals who can encash from it

[@CodeWithAarohi]

So nice of you

[@utkarshtripathi9118]

Ossm video well explained

[@CodeWithAarohi]

Thank you so much

[@Sunil-ez1hx]

Simple awesome . Thank you

[@CodeWithAarohi]

Glad you liked it!

[@soravsingla8782]

Hi Aarohi, your content is excellent and your channel is one of the best Artificial Intelligence channel but still not getting that much of likes which your channel deserves. Hope you succeed #AI #ArtificialIntelligence #DataScience #EducationalContent

[@CodeWithAarohi]

Thank you so much for your kind words and support! It means a lot to me. 😊🙏

[@arnavthakur5409]

Thank you mam for sharing

[@CodeWithAarohi]

Thanks for liking

[@shanikananayakkara4451]

Thank you very much for the amazing knowledge sharing. If you can, please explain how we can use deep unfolding networks for image classification optimisation using a code.

[@CodeWithAarohi]

Sure I will

[@omerkaya2320]

Thank you very much. Please make a video that contains an end to end computer vision project even if the project is basic.

[@CodeWithAarohi]

Sure!

[@pifordtechnologiespvtltd5698]

Really amazing work

[@CodeWithAarohi]

Thank you so much 😀

[@ravindrakarande59]

Please share the dataset used in this video

[@luisaruquipac.381]

Excellent content! Thank you

[@CodeWithAarohi]

Glad you liked it!

[@CodeWithAarohi]

Thank you!

[@jmxt3]

Great video, thanks

[@CodeWithAarohi]

Glad you liked it!

[@rainlarh5306]

Hi Arohi! Thanks for sharing the knowledge:) I have a qns to clarify but I'm not sure whether would you be able to see my comments. How will the the code understand or how was the datasets being seperated into inputs and labels while running the training loop as shown in your video?

[@CodeWithAarohi]

This line is responsible for reading labels and images: datasets.ImageFolder(os.path.join(data_dir, x), data_transforms[x]) for x in ['train', 'val']}

[@Sunil-ez1hx]

Code with Aarohi is best platform to learn Artificial Intelligence & Data Science #BestChannel #CodeWithAarohi

[@karthickkuduva9819]

Mam i tried with my own cnn model including dropout and batch normalization. And i achieved accuracy of 64% and model predicted output label correctly with image. 64% of accuracy is not bad. How to increase accuracy mam ?.

[@CodeWithAarohi]

1- Increase the amount and diversity of your training data. 2- Increase the number of layers (both convolutional and fully connected layers) to capture more complex patterns. 3- Experiment with different hyperparameters like learning rate, optimizers. 4- Use pre-trained models (e.g., VGG, ResNet, Inception) and fine-tune them on your dataset.

[@karthickkuduva9819]

@@CodeWithAarohi thanks for your guidance mam

[@dibo1934]

very helpful video

[@CodeWithAarohi]

Glad it was helpful!

[@MS-yy2dh]

Thank for the video. Can I ask - how do you crate the directory structure with just daisies and dandelions in separate folders? The file I have downloaded (from the link you give) has daisy, dandelion, rose, sunflower and tulip, all together.

[@CodeWithAarohi]

delete rest of the folders

[@harshawithhonor6992]

Hello ma'am, could you please provide the source from where I could get the image files to run this project. Also, do you have any citations (references) for this project.

[@felipemunoz6561]

where i can find that dataset?, i just found of CNN in his github :(

[@CodeWithAarohi]

universe.roboflow.com/enrico-garaiman/flowers-y6mda/dataset/7

[@soravsingla6574]

Very good video

[@CodeWithAarohi]

Thanks

[@andreadotta73]

Hello, great video! I wanted to ask why you used model instead of new_model in the line output = model(input_batch)? new_model should have only 2 neurons in the last layer and therefore choose between two solutions, while model still has all the neurons. Am I correct or am I mistaken? Thanks!!

[@CodeWithAarohi]

Check the cell below "Classification on unseen image". Therewe are loading a pre-trained ResNet-18 model and its saved weights from 'flower_classification_model.pth', then creates a new ResNet-18 model adjusted to classify 2 classes (daisy and dandelion). It copies only the first 2 output units' weights and biases from the loaded model to the final layer of the new model, effectively adapting the pre-trained model for a 2-class problem.

[@andreadotta73]

Okay, thank you! So, load the model with 1000 final nodes and then load our model which has only 2 outputs. Next, we create a new model and copy only the first 2 weights and biases from the initial model. So, to understand, I could directly load the pre-trained model with the exact number of output units, then load my model and use that@@CodeWithAarohi

[@Memorable_days90s]

Can I use flatten() instead of Randomhorizontal()

[@НиколайНовичков-е1э]

Thank you!

[@CodeWithAarohi]

Welcome!

[@commoncats5437]

good work.... do more in Gen ai and LLm's

[@CodeWithAarohi]

Noted!

[@Arceus948]

hey, i m working on an image classifcation project but i m confused what should be the order of preprocessing the images. is my below order of image prepprocessing correct?? step - 1 -> Resizing to 64x64 (Both Train & Validation dataset) step - 2 ->Splitting dataset into train and validation step - 3 ->Augmentation (Only Train data) step - 4 ->Normalization (Both Train & Validation dataset)

[@CodeWithAarohi]

Correct

[@deepakchaudhary3149]

mam if image is of .npy file extension then how to load it?

[@saranshtiwari8543]

x = np.load("x.npy")

[@JohnSmith-gu9gl]

how did you come up with the values: [0.485, 0.456, 0.406] and [0.229, 0.224, 0.225] ?

[@DBWorld]

These values are taken for ImageNet dataset. You need to arrive with your own mean[R,B,G] and std[R,B,G] values for your kind of training dataset.

[@JohnSmith-gu9gl]

@@DBWorld thanks!

[@Thejus_tv]

How can i find that? @DBWorld can you explain?

[@Mehrdadkh87]

Yea

[@gowthamggp4657]

Tq mam

[@CodeWithAarohi]

welcome!

[@nabeelbaig2292]

I have a quick question regarding this video, Aarohi. I watched your video and cloned your GitHub repository to train a dataset of approximately 100 bank cheque images. However, I encountered an issue with the model's performance. When I tested it with non-cheque images, it incorrectly classified them as cheques. On the other hand, it also misclassified bank cheque images as something other than cheques. Can you help me understand and address this problem?

[@CodeWithAarohi]

Imbalanced data can lead to misclassification issues. If you have significantly more cheque images than non-cheque images (or vice versa), it can skew the model's performance. You might need to balance the dataset by oversampling the minority class or undersampling the majority class.

[@krishnazala8735]

can you provide dataset

[@imihhdude]

Can the code snippet apply to multiple labels

[@CodeWithAarohi]

Yes

[@imihhdude]

@@CodeWithAarohi thank you 🫶🏻

[@shaikhyaqoob9386]

how can i get this dataset

[@CodeWithAarohi]

universe.roboflow.com/search?q=flower%20classification

[@mohamedragab8644]

Where is the dataset

[@CodeWithAarohi]

universe.roboflow.com/enrico-garaiman/flowers-y6mda/dataset/7

[@mohamedragab8644]

@@CodeWithAarohi Thank you ❤️

[@Ai_Engineer]

where i can get the datasets

[@CodeWithAarohi]

universe.roboflow.com/enrico-garaiman/flowers-y6mda/dataset/7

[@NabeelBaig-p1y]

Where is Dataset?

[@NabeelBaig-p1y]

Where is Dataset directory?

[@adelilyasgoffa2717]

Thank you, I sent you a mail you didn't answer me,I need your advice please 🙏 , thank you

[@CodeWithAarohi]

Let me check

[@arnavthakur5409]

Keep sharing such an amazing knowledgeable content in form of very easy to learn videos.

[@CodeWithAarohi]

Thank you, I will