Skip to content

sanjaytripathi97/transfer_learning_mobileNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 

Repository files navigation

transfer_learning_mobileNet

step 1:-importing the MobileNet pretrained model and download the weight of mobileNet.
step 1
step 2:-next we are making our layer that will be implmenting on top of the Mobilenet layer.
step 2

step 3:-next we are importing the necessary layers and num_classes here is the category in total,which we are predicting. means how many different catagory we are predicting here i am predicting two values.
step 3
step 4:-next we are doing some image augmentation,generally we do not have the lots of images so by the augmentation we just make new images like by doing zoom,tilt the image etc . a little change in the image for system it is pure new image. here also defining the data location of training and testing or validation dataset. the location in which your data set is present provide the location here.
step 3

step 5:-next we are setting up the epochs , optimizers like RMSprop also saving the model,,in each epochs accuracy is mention the best accuracy will be saved.

step 5

step 6:-here we are fitting the model ans also defining the running accuracy between testing and validation/testing dataset.
step 6

step 7:-next we are simply checking our model that it is predicting right or not.
step 7
step 8
step 9
step 10

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published