Video Demonstration : https://youtu.be/eISIy1pDEFU
It's a task given by Mr. Vimal Daga Sir. In this task I perform Transfer Learning using VGG16 pre-trained weights.
-
- At first I collect the raw images from internet. Then I rename them and also change there format to jpg. Next I classify them and store them in respective folders. This is how I build the dataset.
-
- Then I used the pre-created python file of VGG16 and add my own code to add some new layers to do Transfer Learning.
-
- Then I fit my dataset and start to build the model. Then I save the model.
-
- Lastly I load the saved model and do the prediction.