Skip to content

Latest commit

 

History

History
32 lines (23 loc) · 932 Bytes

README.md

File metadata and controls

32 lines (23 loc) · 932 Bytes
Raw
@@  Editor  :  Horacce
@@  Date    :  2019.03.20

Image Recognition Platform with Baseline Model

This project is for image recognition task using tensorflow, baseline model will be supported with ImageNet pretrained model. Now this project only support SEnet series baseline model training. ReadMe file will be supported later.

To Do: [x] SEnet support [x] multiple gpu support [x] Data Generator Module Optimization [x] VGG [ ] softmax_cross_entropy_with_logit update, disclose the logit, softmax and the CE loss calculation

[ ] ResNet, ResNext [ ] DenseNet [ ] Condense [ ] ReadMe file [ ] voc pretrain & eval [ ] modelfile amelirate (get smaller)

For DL learning and other projects, feel free to subscribe my blog:

Reference:

  1. SEnet-tensorflow: https://github.com/taki0112/SENet-Tensorflow
  2. VGG16: https://github.com/ppplinday/tensorflow-vgg16-train-and-test
  3. ResNet: https://github.com/raghakot/keras-resnet