pythonでのフォルダの読み込みについて

Question

pythonは初心者なので、困っています。
以下はプログラムの一部なのですがこの部分がうまく通りません

path = '/data/0deg_30deg_0811/sleep1'

    # ディレクトリ毎にファイルが何個あるか調べる
    for root, dirs, files in os.walk(path):
        print 'Directory: %s' % root
        nsubFiles = 0
        for f in files:
        #subdir = root.split('/')
        #print '%d' % int(subdir[-1])
            r,e = os.path.splitext(f)
            if e.lower() == '.png':
            #print '%s' % root+'/'+f
            #img = cv2.imread(root+'/'+f)
            #cv2.imshow('result', img)
            #cv2.waitKey(0)
                nFiles += 1
                nsubFiles += 1
        if nsubFiles != 0:
            print '%d files found in %s.' % (nsubFiles, root.split('/')[-1])
    print 'Total: %d files found in %s.' % (nFiles, path)

    # Initialization
    #
    # 訓練画像のサイズ
    height = 80
    width = 40
#    height = 60
#    width = 30
#
#    height = 48
#    width = 48
    # ラベルを入れるための配列
    target = np.zeros(nFiles, dtype=np.int)
    # 訓練画像を入れるための配列
    images = np.zeros((nFiles, height, width), dtype=np.float32)

    # ディレクトリを辿ってラベル、画像を格納する
    nFiles = 0
    for root, dirs, files in os.walk(path):
#        print 'Directory: %s' % root
        subdir = root.split('/')
        if subdir[-1] == 'sleep1':
            continue
        nsubFiles = 0
        for f in files:
            #print '%d' % int(subdir[-1])
            target[nFiles] = int(subdir[-1])-1
            r,e = os.path.splitext(f)
            if e.lower() == '.png':
                #            print '%s' % root+'/'+f
                img = imresize(imread(root+'/'+f), (height,width), interp='bilinear')
                image = np.asarray(img, dtype=np.float32)
                image /= 255.0  # scale uint8 coded colors to the [0.0, 1.0] floats
                images[nFiles, ...] = image
                nFiles += 1
                nsubFiles += 1
        if nsubFiles != 0:
            print '%d files found in %s.' % (nsubFiles, root.split('/')[-1])
    print 'Total: %d files found in %s.' % (nFiles, path)

実行すると

Directory: /data/0deg_30deg_0811/sleep1
1000 files found in sleep1.
Total: 1000 files found in /data/0deg_30deg_0811/sleep1.
Total: 0 files found in /data/0deg_30deg_0811/sleep1.

Traceback (most recent call last):
File "C:\python\NN\mlp_cg.py", line 242, in <module>
data=images.reshape(len(images), -1)
ValueError: total size of new array must be unchanged

と怒られます。そもそもなぜ一つ目のtotalはフォルダの数を数えてくれているのに、次のtotalは0何でしょか。
また
target[nFiles] = int(subdir[-1])-1
は何を示しているんでしょうか

Answer 1

そもそもなぜ一つ目のtotalはフォルダの数を数えてくれているのに、次のtotalは0何でしょうか

以下の通りかと。

• /data/0deg_30deg_0811/sleep1ディレクトリは、ディレクトリ直下に1000個のファイルが存在する。サブディレクトリ以下のファイルは存在しない。
• 最初のループは/data/0deg_30deg_0811/sleep1ディレクトリ以下のファイル数をカウントしている。
  ⇒ 1000
• 次のループは/data/0deg_30deg_0811/sleep1ディレクトリ以下のファイル数をカウントしているが、直下のファイルを処理対象外にしている。
  ⇒ 0

# ディレクトリのrootがsleep1の場合は次のループに移っている。
# つまり、ディレクトリ直下のファイルに対しては以降の処理を行わないということ。
    for root, dirs, files in os.walk(path):
#        print 'Directory: %s' % root
        subdir = root.split('/')
        if subdir[-1] == 'sleep1':
            continue

Comments

[kosuke1126]

処理対象外ですか…
ではどうしたらカウントしてくれるんでしょうか