在pickle数据上无法识别属性重塑

时间:2017-07-18 22:59:18

标签: python tensorflow ipython pickle

我正在为svhn数据集进行数字识别的深度学习模型,我的项目是大约5个ipython笔记本,我的一个笔记本创建了一个pickle文件tensorflow_data.pickle,文件创建成功但是在加载时进入容器名为" dataset"并使用.reshape()我得到了这个错误说: ('无法处理来自dataset.pickle',':',属性错误("'无类型'对象没有属性'重塑& #39;",))

error details

这是我的代码:

try:
with open('tensorflow_data.pickle', 'rb') as f:
    dataset = pickle.load(f)
    train_dataset = dataset['train']['data'].reshape(-1, 32, 32, 1)
    test_dataset = dataset['test']['data'].reshape(-1, 32, 32, 1)
    validation_dataset = dataset['valid']['data'].reshape(-1, 32, 32, 1)
    extra_dataset = dataset['extra']['data'].reshape(-1, 32, 32, 1)
    train_labels = dataset['train']['label']
    test_labels = dataset['test']['label']
    validation_labels = dataset['valid']['label']
    extra_labels = dataset['extra']['label']

except Exception as e:
print('Unable to process data from dataset.pickle', ':', e)
raise

对此有何帮助? PS:我使用Python 2.7并在ubuntu上通过anaconda安装了tensorflow

0 个答案:

没有答案