質問編集履歴
1
パス変更
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -30,8 +30,8 @@
|
|
|
30
30
|
|
|
31
31
|
epoch = 2
|
|
32
32
|
|
|
33
|
-
result_dir = 'D:/
|
|
33
|
+
result_dir = 'D:/result'
|
|
34
|
-
if not os.path.exists(
|
|
34
|
+
if not os.path.exists(result):
|
|
35
35
|
os.mkdir(result_dir)
|
|
36
36
|
|
|
37
37
|
|
|
@@ -69,7 +69,7 @@
|
|
|
69
69
|
|
|
70
70
|
|
|
71
71
|
if __name__ == '__main__':
|
|
72
|
-
classes = ['
|
|
72
|
+
classes = ['m', 'n', 'o', 'k']
|
|
73
73
|
|
|
74
74
|
batch_size = 32
|
|
75
75
|
nb_classes = len(classes)
|
|
@@ -110,7 +110,7 @@
|
|
|
110
110
|
test_datagen = ImageDataGenerator(rescale=1.0 // 255)
|
|
111
111
|
|
|
112
112
|
train_generator = train_datagen.flow_from_directory(
|
|
113
|
-
directory='D:/
|
|
113
|
+
directory='D:/s_dir/train_images',
|
|
114
114
|
target_size=(img_rows, img_cols),
|
|
115
115
|
color_mode='rgb',
|
|
116
116
|
classes=classes,
|
|
@@ -122,7 +122,7 @@
|
|
|
122
122
|
print(train_generator.class_indices)
|
|
123
123
|
|
|
124
124
|
test_generator = test_datagen.flow_from_directory(
|
|
125
|
-
directory='D:/
|
|
125
|
+
directory='D:/s_dir/test_images',
|
|
126
126
|
target_size=(img_rows, img_cols),
|
|
127
127
|
color_mode='rgb',
|
|
128
128
|
classes=classes,
|
|
@@ -141,7 +141,7 @@
|
|
|
141
141
|
validation_steps=nb_val_samples,
|
|
142
142
|
callbacks=[checkpointer])
|
|
143
143
|
|
|
144
|
-
save_history(history, os.path.join(
|
|
144
|
+
save_history(history, os.path.join(result, 'history.txt'))
|
|
145
145
|
|
|
146
146
|
# modelに学習させた時の変化の様子をplot
|
|
147
147
|
plot_history(history)
|