pythonで自作データセットを作成したい

Question

前提・実現したいこと

pythonで自分の画像を用いてデータセットを作成しようと思っています。

いろいろ調べたところ、同じような事をしている記事があったのでそれを参考に実行したのですが上手くできませんでした

発生している問題

下記ソースコードの

for picture in list_pictures('./dataset1/'):
    img = img_to_array(load_img(picture, target_size=(64,64)))
    X.append(img)
    Y.append(0)

の部分が実行されておらず、X、Yが空になっている。
そのため、

X_train, X_test, y_train, y_test = train_test_split(X, Y, test_size=0.33, random_state=111)

でエラーになっている。

エラーコード

ValueError: With n_samples=0, test_size=0.2 and train_size=None, the resulting train set will be empty. Adjust any of the aforementioned parameters.

該当のソースコード

import keras
from keras.utils import np_utils
from keras.layers.convolutional import Conv2D, MaxPooling2D
from keras.models import Sequential
from keras.layers.core import Dense, Dropout, Activation, Flatten
from keras.preprocessing.image import array_to_img, img_to_array, load_img
import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split
import matplotlib.pyplot as plt

import os
import re


def list_pictures(directory, ext='jpg|jpeg|bmp|png|ppm'):
    return [os.path.join(root, f)
            for root, _, files in os.walk(directory) for f in files
            if re.match(r'([\w]+.(?:' + ext + '))', f.lower())]

X = []
Y = []

for picture in list_pictures('./dataset1/'):
    img = img_to_array(load_img(picture, target_size=(64,64)))
    X.append(img)
    Y.append(0)

X = np.asarray(X)
Y = np.asarray(Y)


X = X.astype('float32')
X = X / 255.0


Y = np_utils.to_categorical(Y, 2)


X_train, X_test, y_train, y_test = train_test_split(X, Y, test_size=0.33, random_state=111)

試したこと

上記のソースコードを実行してdateset1の中の画像でデータセットを作成しようとしたがうまくできませんでした。

おそらく

def list_pictures(directory, ext='jpg|jpeg|bmp|png|ppm'):
    return [os.path.join(root, f)
            for root, _, files in os.walk(directory) for f in files
            if re.match(r'([\w]+.(?:' + ext + '))', f.lower())]

この部分が原因だとは思うのですが、解決方法が分かりませんでした。

参考サイト
https://qiita.com/AlphaMikeNeko/items/9870533f3ab1e11e340f

Comments

[jbpb0]

pythonのコードの一番最初の行のすぐ上に
```python
だけの行を追加してください
また、pythonのコードの一番最後の行のすぐ下に
```
だけの行を追加してください

[jbpb0]

OSは何でしょうか?

[takesiiii]

変更しました。ご指摘ありがとうございます。

OSはwindowsです。

[jbpb0]

フォルダ名だけ変えて実行したら、正常に動きました
こちらもWindowsです

for picture in list_pictures(...
のループ内のどこかに
print(picture)
を追加してみてください (行頭にはインデントを入れてください)
正常ならば、
./dataset1/画像ファイル名
が、画像ファイルの数だけ表示されるはずですが、どうでしょうか?

[jbpb0]

画像ファイル名にスペースや「-」(ハイフン)や「.」(ピリオド)が入ってるとダメです
ご確認ください (他にもダメな記号あるかもしれない)

「.」(ピリオド)は、拡張子(jpgとか)の前だけは、もちろん大丈夫です

[jbpb0]

「(」「)」もダメでした

もし、dataset1フォルダに置いてある画像のファイル名に、何かひっかかりそうな文字を使っているなら、「a.jpg」みたいな絶対大丈夫なファイル名の画像ファイルを試しに置いてみて、プログラム実行してみてください

[takesiiii]

画像のファイル名を変更したら実行出来ました。
ありがとうございます。

Answer 1

画像ファイル名にスペースや「-」(ハイフン)や「.」(ピリオド)や「(」「)」が入ってるとダメです
ご確認ください (他にもダメな記号あるかもしれない)
もし、dataset1フォルダに置いてある画像のファイル名に、何かひっかかりそうな文字を使っているなら、「a.jpg」みたいな絶対大丈夫なファイル名の画像ファイルを試しに置いてみて、プログラム実行してみてください

なお、「.」(ピリオド)は、拡張子(jpgとか)の前だけは、もちろん大丈夫です