您的位置:首页 > 其它

运行tensorflow cifar10教程时,cifar-10-binary.tar.gz下载太慢,自己下载并解压方法

2018-01-11 17:15 435 查看
在做Tensorflow官方文档cifar10,运行cifar10_train.py时,网速太慢,三个小时只下载了30%,于是找方法自己下载并解压。

步骤如下:

先下载文件(包括cifar-10-binary.tar.gz 和 用到的cifar10.py cifar10_train.py等文件)

链接:https://pan.baidu.com/s/1eTwtvvc 密码:deky

将cifar-10-binary.tar.gz拷贝到/tmp/cifar10_data/。如果之前运行过程序,需要将此路径下的cifar-10-batches-bin文件夹删除。

具体:

在终端进入下载文件的目录,之后

cp cifar-10-binary.tar.gz /tmp/cifar10_data/

rm -r /tmp/cifar10_data/cifar-10-batches-bin

之后,ls /tmp/cifar10_data,检查是否只有cifar-10-binary.tar.gz一个文件

确认无误后,下载cifar10.py and cifar10_input.py,在同路径下运行代码:

#!/usr/bin/python
#-*-coding:utf-8-*-

import cifar10
import cifar10_input
import tensorflow as tf
import os
import tarfile

#解压缩
filepath = '/tmp/cifar10_data/cifar-10-binary.tar.gz'
dest_directory = '/tmp/cifar10_data'
extracted_dir_path = os.path.join(dest_directory, 'cifar-10-batches-bin')
if not os.path.exists(extracted_dir_path):
tarfile.open(filepath, 'r:gz').extractall(dest_directory)

data_dir = '/tmp/cifar10_data/cifar-10-batches-bin'
batch_size = 100

#生成CIFAR-10的训练数据和训练标签数据
images_train, labels_train = cifar10_input.distorted_inputs(data_dir=data_dir, batch_size=batch_size)

#生成CIFAR-10的测试数据和测试标签数据
images_test, labels_test = cifar10_input.inputs(eval_data=True, data_dir=data_dir, batch_size=batch_size)

sess = tf.InteractiveSession()
tf.global_variables_initializer()
tf.train.start_queue_runners()
print(images_train)
print(images_test)


代码运行结束后,可以再进入目录看一下,可以看到将cifar10数据集成功解压啦~

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息