2016-03-30 25 views
0

我目前正在追蹤CIFAR-10的例子。已下載的數據集位於http://www.cs.toronto.edu/~kriz/cifar.html(CIFAR-10 python版本)。提取文件後,「cifar-10-batches-py」文件夾出來了。關於運行CIFAR-10數據集的cPickle庫的問題

有了這個文件夾,我試着看到它的數據集。我使用下面顯示的代碼。

import cPickle 
import os 
import numpy as np 

def unpickle(file): 
    fo = open(file, 'rb') 
    dict = cPickle.load(fo) 
    fo.close() 
    return dict 

def conv_data2image(data): 
    return np.rollaxis(data.reshape((3,32,32)),0,3) 

def get_cifar10(folder): 
    tr_data = np.empty((0,32*32*3)) 
    tr_labels = np.empty(1) 
    ''' 
    32x32x3 
    ''' 
    for i in range(1,6): 
     fname = os.path.join(folder, "%s%d" % ("data_batch_", i)) 
     data_dict = unpickle(fname) 
     if i == 1: 
      tr_data = data_dict['data'] 
      tr_labels = data_dict['labels'] 
     else: 
      tr_data = np.vstack((tr_data, data_dict['data'])) 
      tr_labels = np.hstack((tr_labels, data_dict['labels'])) 

    data_dict = unpickle(os.path.join(folder, 'test_batch')) 
    te_data = data_dict['data'] 
    te_labels = np.array(data_dict['labels']) 

    bm = unpickle(os.path.join(folder, 'batches.meta')) 
    label_names = bm['label_names'] 

    return tr_data, tr_labels, te_data, te_labels, label_names 


if __name__ == '__main__': 
    datapath = '/Users/sungtegg/Documents/cifar-10-batch.py' 

    tr_data10, tr_labels10, te_data10, te_labels10, label_names10 = get_cifar10(datapath) 

編譯此代碼後,

"IOError: [Errno 2] No such file or directory: '/Users/sungtegg/Documents/cifar-10-batch.py/data_batch_1' 

此錯誤代碼就出來了。

哪段代碼出錯了?

+0

的代碼在http://qiita.com/supersaiakujin/items/5e9d2b2850e256f99982所示 – teggme

回答

1

您的數據路徑不正確,您可以從CIFAR10下載並解壓縮到任何文件夾。確保你指向正確的數據集目錄。

數據路徑= '/用戶/ sungtegg /文檔/ cifar10-數據集'