Google官方给的cifar10源码是基于很早的版本的tensorflow,目前tensorflow已更新到1.4.0版本.所以源码很多不适合.需要修改很多弃用的函数.
google官方的源码
训练数据官网
修改后的代码,适用tensorflow1.4.0:链接: https://pan.baidu.com/s/1hsAJBBu 密码: ynwn
训练数据百度云:链接: https://pan.baidu.com/s/1eSJBn7G 密码: h958
安装tensorflow方法推荐docker安装,方便快捷
下载上面给出的源代码和训练数据
然后使用:python cifar10_train.py
开始训练
源代码中注释了从国外下载训练数据的代码,所以首次运行会报错找不到训练数据的文件.根据报错的路径将下载的训练数据放到具体路径下即可.一般的默认路径是/tmp/cifar10_data/cifar-10-batches-bin/
将解压出来的那6个.bin结尾的文件放到这个目录下即可
然后python cifar10_train.py开始训练
大致返回结果如图:
![1E[2ZEJ56G478DM$1]]4EJ.png
警告信息忽略即可
网友评论