Skip to content

jjdblast/Fine_Grained_Classification

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

55 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Fine_Grained_Classification

Task : Classification of 196 classes of cars with less than 9k images for training. It's intend for replication of the work :

Monza: Image Classification of Vehicle Make and Model Using Convolutional Neural Networks and Transfer Learning
  1. use pre-trained googlenet
  2. train with normal rates
  3. Achieved 87% top1 accuracy and 97% top5 acc with 10000 iterations

Lessons learned :

1.  Training from scratch is very hard with less data and more class.
2.  Bounding box of the car really helps
3.  fine tuning need to select a good lr and make the bottom layers learn slow, but the last few layers learn fast.

About

Fined grained classification On Car dataset

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 57.7%
  • C++ 33.3%
  • Python 4.1%
  • Cuda 2.5%
  • CMake 1.2%
  • MATLAB 0.4%
  • Other 0.8%