Chainerのデータセットの作り方

Question

前提・実現したいこと

現在、chainerを用いて画像分類をするために、chainer.datasets.ImageDatasetを使おうと考えていますが、引数に何を指定すればいいのかわからない状態です。

現在用意してあるデータは、約20000枚の画像データと、それらの画像をOpenCVを用いてnumpyの形に保存してあるファイル（.npzのファイル)と.npzファイルに対応するようにラベルを与えた配列があります。

これらのデータをどのようにchainer.datasets.ImageDatasetに与えたら良いのか教えてください。

発生している問題・エラーメッセージ

エラーメッセージ

該当のソースコード

"""
validationディレクトリにある画像をnumpyに変換し、ラベルの配列を用意するコード
"""
import os
import numpy as np
import cv2

os.chdir("./validation")
cWD = os.getcwd()
filenames = os.listdir(cWD)
num_files = len(filenames)

h = 100                                                 #画像の高さ
w = 100                                                 #画像の幅
data = np.zeros((num_files,3,h,w))                      #画像を入れる箱を用意

print(data.shape)
for i,filename in zip(range(num_files),filenames):      #validationディレクトリのファイルとiを同時に回す

        img = cv2.imread(filename,1)
        one_data = img.reshape(3,h,w)                   #100*100*3を3*100*100に変換
        data[i] = one_data                              #dataに読み込んだ画像（変換済み）を入れる
u = np.zeros((num_files),dtype='int32')

for i,filename in zip(range(num_files),filenames):
        positive_or_negative = filename.split('e',1)    #imageとnegaimageを区別する（"imag"e,"n"egaimage)

        if positive_or_negative[0]=="imag":             #positiveの画像にラベルとして0を与える
                u[i] = 0
        if positive_or_negative[0]=="n":                #negativeの画像にラベルとして1を与える
                u[i] = 1

os.chdir("../")
np.savez('validation.npz',x=data,t=u)                   #data(画像)をｘに、u(ラベル)をｔとして保存する

試したこと

ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

例えば、ここ。

画像ファイルのファイル名と、そのラベルが書かれたテキストファイルを用意して、LabeledImageDataSetにそのテキストファイルへのパス名を渡してやればよい。PILがいるので、インストールを忘れずに。

Comments

[nematoda]

解答ありがとうございます。
LabeledimageDataSetを使ってChainerを動かすことができました。

Answer 2

ベストアンサーに選ばさせていただいた方の解答の情報を基にchainer.datasets.ImageDatasetにテキストファイルを渡すことで解決しました。

以下のコードは画像ファイルのパス情報と対応するラベルをテキストファイルに出力したコードになります。
一部、過去の自分のコードを使いまわした部分があるので、余計な箇所があると思います。

import os
import numpy as np
import cv2

os.chdir("./validation")
cWD = os.getcwd()
filenames = os.listdir(cWD)
num_files = len(filenames)

h = 100                                                 #画像の高さ
w = 100                                                 #画像の幅


u =np.zeros((num_files),dtype='int32')

os.chdir("../")
f = open('images.txt','w')


for i,filename in zip(range(num_files),filenames):      #validationディレクトリのファイルとiを同時に回す


        positive_or_negative = filename.split('e',1)    #imageとnegaimageを区別する（"imag"e,"n"egaimage)

        if positive_or_negative[0]=="imag":             #positiveの画像にラベルとして0を与える
                u[i] = 0
        if positive_or_negative[0]=="n":                #negativeの画像にラベルとして1を与える
                u[i] = 1

        f.write("./validation/"+filename + " " +str(u[i]) +" \n")