KO | EN

GitLite

Minimalist Open Source Repository Search Engine

gitlite — search

#dsh #dsh-plugin #typescript #developer-tools #claude-code #ai-agent #codex #deepseek-harness #ai-agents #cli #react #mcp

keras-dogs

★ 115 Open GitHub ↗

a baseline for baidu dog classification competition.

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Fine grain Dog Classification held by Baidu

> author: [cweihang](https://github.com/ahangchen)

Star this repository if you find it helpful, thank you.

Language: [English](README.md)/[简体中文](README_cn.md)
## About
This is a dog classification competition held by Baidu. Competition URL: http://js.baidu.com/

## Framework
- [Keras](https://keras.io/)
- [Tensorflow Backend](https://www.tensorflow.org/)

## Hardware
- Geforce GTX 1060 6G
- Intel® Core™ i7-6700 CPU
- Memory 8G

## Model
- [Xception](https://arxiv.org/abs/1610.02357) for deep feature extraction

[Structure of Xception](doc/large_img.md##Xception)

- Category loss plus Binary loss inspired by [the idea in Person Re-id](https://arxiv.org/abs/1611.05666)
![](https://raw.githubusercontent.com/ahangchen/keras-dogs/master/viz/re-id-combined-loss.png)

### Implemented in Keras
- Remove the final classify dense layer in Xception to get the deep feature
- Input two images, containing same or different labels
- Train the model with categorical loss of two images and their class labels
- Train the model with binary loss meaning whether two images belong to same class or not

![](https://raw.githubusercontent.com/ahangchen/keras-dogs/master/viz/model_combined.png)

## Data pre-process
- Download the images from Baidu Cloud
  - Training Set: http://pan.baidu.com/s/1slLOqBz Key: 5axb
  - Test set: http://pan.baidu.com/s/1gfaf9rt Key：fl5n
  
> Previous link was provided by Baidu and only Baidu has the rights to spread the data, thus I can not provide the data anymore if Baidu canceled the link shared. You can refer to http://dianshi.baidu.com/gemstone/competitions/detail?raceId=17 with another newly published dataset on which you can apply the same algorithm. You can also apply this model on Person ReID dataset [Market1501](http://www.liangzheng.org/Project/project_reid.html). But you should do some processing on the data. You can refer to part of our CVPR paper code: [rank-reid](https://github.com/ahangchen/rank-reid)

- Place the images with the same class into same directory, for using ImageDataGenerator.
- Because I named the images with the format "typeid_randhash.jpg", I wrote [img2keras.py](preprocess/img2keras.py) for the work described above.
- There are more details to handle. If you meet any error, refer the Keras document first. If you still have some question, you can create an [issue](https://github.com/ahangchen/keras-dogs/issues).

## Training
- Use ImageDataGenerator for data argumentation
- It's hard to find positive samples for binary training using ImageDataGenerator, because the samples are shuffled.
Looking throughout the data set for positive samples is inefficient. Fortunately, in each batch, we can find some samples with the same class.
So we simply swap those samples to construct positive samples.
- Frozen the Xception CNN layers, train the full connected layers for category and binary classification with ADAM
- Unfroze final two blocks(layer 105 to the end) of Xception, continue training with SGD
- Remove data argumentation, retrain until converge

## Code
- Single Xception Model
  - Train: [single_model.py](single/single_model.py)
  - Test: [single_model_test.py](single/single_model_test.py)
- Multi loss Model
  - Froze and fine-tuning: [froze_fine_tune.py](xception/froze_fine_tune.py)
  - Fine tune with some trick: [trick_tune.py](xception/trick_tune.py)
  - Test: [baidu_dog_test.py](xception/baidu_dog_test.py)

## Result
- InceptionV3, softmax loss: 0.2502
- Xception, softmax loss: 0.2235
- Xception, multi loss: 0.211
- Xception, multi loss, fine tune without data argumentation: 0.2045

> If you find some bug in this code, create an issue or a pull request to fix it, thanks!

keras-dogs

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README