Changed some values

Author: BergerAl

README.md

@@ -1,3 +1,7 @@
 # face-detection-on-raspberry-pi
 
 Using OpenCV2 for Face Detection. Runs on Raspberry Pi model B. But takes ~10 seconds with haarcascade.
+
+Model was trained with train_CNN on /data. The result was loss: 0.0926 - acc: 0.9649 - val_loss: 0.0657 - val_acc: 0.9811.
+
+By training with not filtered images /old_data. The result was loss: 0.2127 - acc: 0.9066 - val_loss: 0.4971 - val_acc: 0.8152. Because of this result a preprocessing of the taken image should be used.

models/basic_cnn_20_epochs.h5

@@ -9,10 +9,10 @@
 
 img_width, img_height = 150, 150
 
-train_data_dir = 'data/train'
-validation_data_dir = 'data/validation'
+train_data_dir = 'old_data/train'
+validation_data_dir = 'old_data/validation'
 
-classes_amount = 5
+classes_amount = 4
 
 # used to rescale the pixel values from [0, 255] to [0, 1] interval
 datagen = ImageDataGenerator(
@@ -73,8 +73,8 @@
 
 batch_size = 16
 nb_epoch = 30
-nb_train_samples = 924#96*5                # old_data == 283,data ==214
-nb_validation_samples = 360#48*5            # old_data == 38,data ==27
+nb_train_samples = 283*4                # old_data == 283,data ==214
+nb_validation_samples = 38*4            # old_data == 38,data ==27
 
 model.fit_generator(
         train_generator,
@@ -83,4 +83,4 @@
         validation_data=validation_generator,
         validation_steps=nb_validation_samples / batch_size)
 
-model.save_weights('models/basic_cnn_30_epochs.h5')
+model.save_weights('models/basic_cnn_30_epochs_old_data.h5')