Skip to content

Arunangshu-Nandy/CNN-with-Transfer-Learning-for-Animal-Images

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

In this transfer learning case, I have created my own database of some animal images (such as tiger, snow leopard, leopard & elephant) and trained the dataset with help of tensorflow mobilenet V2 model for image classification. The mobilenet model has been pre-trained over millions of images. I have used the same feature vector to train my database. CNN (convoluted neural network) along with Max Pooling has been used in the feature vector training.

Mobilenet V2 model from tensorflow hub link- https://tfhub.dev/google/tf2-preview/mobilenet_v2/classification/4