3

theano logistic_sgd.py 実装で提供される mnist.pkl.gz に似たデータセットを構築しようとしています。以下は私のコードスニペットです。

import numpy as np
import csv
from PIL import Image
import gzip, cPickle
import theano
from theano import tensor as T

def load_dir_data(csv_file=""):
    print(" reading: %s" %csv_file)
    dataset=[]
    labels=[]

    cr=csv.reader(open(csv_file,"rb"))
    for row in cr:
        print row[0], row[1]
        try: 
            image=Image.open(row[0]+'.jpg').convert('LA') 
            pixels=[f[0] for f in list(image.getdata())]
            dataset.append(pixels)
            labels.append(row[1])
            del image 
        except: 
            print("image not found")
    ret_val=np.array(dataset,dtype=theano.config.floatX)
    return ret_val,np.array(labels).astype(float)   


def generate_pkl_file(csv_file=""):
    Data, y =load_dir_data(csv_file)
    train_set_x = Data[:1500]
    val_set_x = Data[1501:1750]
    test_set_x = Data[1751:1900]
    train_set_y = y[:1500]
    val_set_y = y[1501:1750]
    test_set_y = y[1751:1900]
    # Divided dataset into 3 parts. I had 2000 images.

    train_set = train_set_x, train_set_y
    val_set = val_set_x, val_set_y
    test_set = test_set_x, val_set_y

    dataset = [train_set, val_set, test_set]

    f = gzip.open('file.pkl.gz','wb')
    cPickle.dump(dataset, f, protocol=2)
    f.close()    


if __name__=='__main__':
    generate_pkl_file("trainLabels.csv")

エラー メッセージ: トレースバック (最新の呼び出しが最後):

  File "convert_dataset_pkl_file.py", line 50, in <module>
    generate_pkl_file("trainLabels.csv") 
  File "convert_dataset_pkl_file.py", line 29, in generate_pkl_file
    Data, y =load_dir_data(csv_file)
  File "convert_dataset_pkl_file.py", line 24, in load_dir_data
    ret_val=np.array(dataset,dtype=theano.config.floatX)
ValueError: setting an array element with a sequence.

csvファイルには2つのフィールドが含まれています..画像名、Pythonインタープリターでこれを実行すると分類ラベル、それは私のために働いているようです..次のように..ここで配列要素をシーケンスで設定するというエラーが発生しません..

----------python インタプリタ出力----------

image=Image.open('sample.jpg').convert('LA')
pixels=[f[0] for f in list(image.getdata())]
dataset=[]
dataset.append(pixels)
dataset.append(pixels)
dataset.append(pixels)
dataset.append(pixels)
dataset.append(pixels)
b=numpy.array(dataset,dtype=theano.config.floatX)
b
array([[ 2.,  0.,  0., ...,  0.,  0.,  0.],
       [ 2.,  0.,  0., ...,  0.,  0.,  0.],
       [ 2.,  0.,  0., ...,  0.,  0.,  0.],
       [ 2.,  0.,  0., ...,  0.,  0.,  0.],
       [ 2.,  0.,  0., ...,  0.,  0.,  0.]])

同じ一連の命令を (論理的に) 実行しているにもかかわらず、sample.py を実行すると、valueError: 配列要素をシーケンスで設定する..この動作を理解しようとしています..どんな助けも素晴らしいでしょう..

4

1 に答える 1

6

問題はおそらくこの質問と似ています。

画像ごとに行を持つピクセル値のマトリックスを作成しようとしています。ただし、各画像のサイズは異なるため、各行のピクセル数は異なります。

numpy で「ギザギザ」の float 型の配列を作成することはできません。すべての行は同じ長さでなければなりません。

各行を最大の画像の長さにパディングする必要があります。

于 2015-05-29T08:21:16.367 に答える