本代码主要取自TensorFlow实践5.3节。
博主主要是对自己在实现此代码的过程中遇到的问题进行说明,以及对源代码增加了一定的注释,若有任何不当之处,烦请指教。
在此次实践中,所采用的数据集是CIFAR-10,其中包含了60000张32x32的彩色图像(训练集:50000张,测试集:10000张)。其中标注为10类,每一类6000张,分别是以下十类:airplane、automobile、bird、cat、deer、dog、frog、horse、ship、truck。
在开始编辑代码前,先将本次实践所需要的cifar10.py和cifar10_input.py下载下来(可参考下面两种下载方式),并将拷贝到所建工程的目录下。如下图。
博主提供两种下载方式。
第一种。可通过网址 https://github.com/tensorflow/models/tree/master/tutorials/image/cifar10,文件cifar10.py主要用于下载和解压cifar10数据集,由于此处的两个文件是原始的,因此其设置的filepath对于我们的使用不太方便,可自行设置,博主一般将下载和解压地址设置在当前目录下。即如下更改cifar10.py中第381行代码。
dest_directory = './tmp/cifar10_data'
第二种。可通过博主的云盘下载,链接:https://pan.baidu.com/s/11N9ssULWgiYrKsXdqxXQSg 密码:nvsb,cifar10.py文件中的路径已修改好,可直接使用。
接下来创建主文件,以下代码便可正常使用了。
#####################################################################################################