dengpingling / Fine_Grained_Classification

Fined grained classification On Car dataset

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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


Languages

Language:Jupyter Notebook 57.7%Language:C++ 33.3%Language:Python 4.1%Language:Cuda 2.5%Language:CMake 1.2%Language:MATLAB 0.4%Language:Shell 0.3%Language:Makefile 0.3%Language:CSS 0.1%Language:HTML 0.1%Language:Dockerfile 0.0%