Skip to content

Classification of 120 breeds with DNN and inception blocks

Notifications You must be signed in to change notification settings

Yoskutik/CV-InceptionNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Dogs breed classification

This dataset contains images of 120 breeds of dogs from around the world. Total number of images is 20,580.


The prediction model consist of pre-trained MobileNetV3 model, 2 extra inception blocks and 1 short connection. That's how the model looks like:

Model scheme


The model is saved in best_model.h5 file.
Test accuracy of the model is 0.946, top 2 accuracy is 0.9764