Skip to content

duanqiu/Dog-Breed-Identification-Gluon

 
 

Repository files navigation

Dog-Breed-Identification-Gluon

Kaggle 120种狗分类比赛 Gluon实现代码
通过Gluon教程,边学边练。使用renset152_v1inception_v3进行模型融合得到0.27760分数

详细说明参见Gluon炼丹(Kaggle 120种狗分类,迁移学习加双模型融合)

杨培文的代码中学到的,如果使用imagenet网络进行预训练,锁住特征层的话,那可以先把所有数据都都过一边特征层网络,这样在后面进行分类输出网络的训练时会省去很多时间和显存,并且在训练的时候可以把bs开到很大,而且在过特征层网络的时候,可以把bs设的小一点,而且不用反向传播,速度也不会太慢,这样就解决了在低配电脑上无法训练的问题。而且用最新版mxnet输出的特征数据非常小,70M左右,后面的分类层即使用cpu也能跑很顺畅

About

Kaggle 120种狗分类,Gluon实现

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • CSS 71.1%
  • JavaScript 27.1%
  • Python 1.5%
  • HTML 0.3%