爬取汽车图片,然后使用resnet-50训练汽车的分类器。车辆总共1062类。
- 汽车图片来自易车网。
- 使用了R version 3.4.3
- python3.5.2
- 安装了GPU版本的mxnet,mxnet的相关教程可以访问mxnet教程.
-
pictures_prepare.R是一个图片爬取,存储以及整理的过程,按照要求修改path即可。最后需要调用./mxnet/tools/im2rec.py生成lst和rec文件。
-
train.ipynb是模型训练的过程。
-
predict.py是模型预测的过程。