From 1aeb9bf1fa47126308fbe2bce3e495d79b4ef023 Mon Sep 17 00:00:00 2001 From: Kamila Bobkowska Date: Sun, 10 May 2020 16:25:13 +0000 Subject: [PATCH] touch ups --- Report_Kamila_Bobkowska.md | 20 +++++++++++++++++--- 1 file changed, 17 insertions(+), 3 deletions(-) diff --git a/Report_Kamila_Bobkowska.md b/Report_Kamila_Bobkowska.md index a96bd73..3ad7190 100644 --- a/Report_Kamila_Bobkowska.md +++ b/Report_Kamila_Bobkowska.md @@ -32,15 +32,29 @@ classifier.compile(optimizer = "adam", loss = "binary_crossentropy", metrics = [ ``` After that I did some work with preprocessing the images I was working with. Made sure that the size is still (110, 110). set a batch size, allowed for flips and such. +``` +train_datagen = ImageDataGenerator( + rescale=1./255, + shear_range=0.1, + zoom_range=0.1, + width_shift_range=0.1, + height_shift_range=0.1, + horizontal_flip=True, + vertical_flip=True, +) +``` Later I trained my classifier and saved it's weights for future use in the project. I attempted this about 5 times with different number of epochs and the size per epoch. + +**UPDATE:** +I tried running it once again with more epochs and steps per epoch and get a better resulat and I think a better validation loss than before. ``` -classifier.fit_generator( train_generator, steps_per_epoch = 150, epochs = 25, validation_data = test_generator ) -classifier.save_weights('model_ver_4.h5') +classifier.fit_generator( train_generator, steps_per_epoch = 165, epochs = 32, validation_data = test_generator ) +classifier.save_weights('model_ver_5.h5') ``` Training: -![Example](https://i.imgur.com/FNoNLJq.png) +![Example](https://i.imgur.com/HBBvWCM.png) ## Changes in the common part