回答編集履歴
1
d
test
CHANGED
@@ -1,3 +1,7 @@
|
|
1
|
+
[公式リファレンス](https://keras.io/ja/preprocessing/image/) を確認するのが一番よいです。
|
2
|
+
|
3
|
+
|
4
|
+
|
1
5
|
Keras の flow_from_directory は、以下のようなフォルダ構成を仮定しています。
|
2
6
|
|
3
7
|
各サブフォルダが1クラスに該当するので、そのクラスの画像を入れておくと読み込んだ際に flow_from_directory() 内でラベル付けしてくれます。
|
@@ -23,3 +27,75 @@
|
|
23
27
|
└─bird
|
24
28
|
|
25
29
|
```
|
30
|
+
|
31
|
+
|
32
|
+
|
33
|
+
```python
|
34
|
+
|
35
|
+
from tensorflow.python.keras.preprocessing.image import ImageDataGenerator
|
36
|
+
|
37
|
+
|
38
|
+
|
39
|
+
train_datagen = ImageDataGenerator(rescale=1./255)
|
40
|
+
|
41
|
+
|
42
|
+
|
43
|
+
train_generator = train_datagen.flow_from_directory(
|
44
|
+
|
45
|
+
'dataset',
|
46
|
+
|
47
|
+
target_size=(224, 224), # モデルの入力サイズ
|
48
|
+
|
49
|
+
color_mode='grayscale', # 読み込み形式
|
50
|
+
|
51
|
+
batch_size=20, # バッチサイズ
|
52
|
+
|
53
|
+
class_mode='categorical')
|
54
|
+
|
55
|
+
```
|
56
|
+
|
57
|
+
|
58
|
+
|
59
|
+
例: 良品、不良品の場合
|
60
|
+
|
61
|
+
```
|
62
|
+
|
63
|
+
dataset
|
64
|
+
|
65
|
+
├─positive: 不良品の画像
|
66
|
+
|
67
|
+
│ ├─1.png
|
68
|
+
|
69
|
+
│ ├─2.png
|
70
|
+
|
71
|
+
│ ...
|
72
|
+
|
73
|
+
└─negative: 良品の画像
|
74
|
+
|
75
|
+
```
|
76
|
+
|
77
|
+
|
78
|
+
|
79
|
+
```python
|
80
|
+
|
81
|
+
from tensorflow.python.keras.preprocessing.image import ImageDataGenerator
|
82
|
+
|
83
|
+
|
84
|
+
|
85
|
+
train_datagen = ImageDataGenerator(rescale=1./255)
|
86
|
+
|
87
|
+
|
88
|
+
|
89
|
+
train_generator = train_datagen.flow_from_directory(
|
90
|
+
|
91
|
+
'dataset',
|
92
|
+
|
93
|
+
target_size=(224, 224), # モデルの入力サイズ
|
94
|
+
|
95
|
+
color_mode='grayscale', # 読み込み形式
|
96
|
+
|
97
|
+
batch_size=20, # バッチサイズ
|
98
|
+
|
99
|
+
class_mode='binary')
|
100
|
+
|
101
|
+
```
|