teratail header banner
teratail header banner
質問するログイン新規登録

回答編集履歴

1

d

2019/02/26 14:18

投稿

tiitoi
tiitoi

スコア21960

answer CHANGED
@@ -1,3 +1,5 @@
1
+ [公式リファレンス](https://keras.io/ja/preprocessing/image/) を確認するのが一番よいです。
2
+
1
3
  Keras の flow_from_directory は、以下のようなフォルダ構成を仮定しています。
2
4
  各サブフォルダが1クラスに該当するので、そのクラスの画像を入れておくと読み込んだ際に flow_from_directory() 内でラベル付けしてくれます。
3
5
 
@@ -10,4 +12,40 @@
10
12
  │ ...
11
13
  ├─dog
12
14
  └─bird
15
+ ```
16
+
17
+ ```python
18
+ from tensorflow.python.keras.preprocessing.image import ImageDataGenerator
19
+
20
+ train_datagen = ImageDataGenerator(rescale=1./255)
21
+
22
+ train_generator = train_datagen.flow_from_directory(
23
+ 'dataset',
24
+ target_size=(224, 224), # モデルの入力サイズ
25
+ color_mode='grayscale', # 読み込み形式
26
+ batch_size=20, # バッチサイズ
27
+ class_mode='categorical')
28
+ ```
29
+
30
+ 例: 良品、不良品の場合
31
+ ```
32
+ dataset
33
+ ├─positive: 不良品の画像
34
+ │ ├─1.png
35
+ │ ├─2.png
36
+ │ ...
37
+ └─negative: 良品の画像
38
+ ```
39
+
40
+ ```python
41
+ from tensorflow.python.keras.preprocessing.image import ImageDataGenerator
42
+
43
+ train_datagen = ImageDataGenerator(rescale=1./255)
44
+
45
+ train_generator = train_datagen.flow_from_directory(
46
+ 'dataset',
47
+ target_size=(224, 224), # モデルの入力サイズ
48
+ color_mode='grayscale', # 読み込み形式
49
+ batch_size=20, # バッチサイズ
50
+ class_mode='binary')
13
51
  ```