From be9da6c91f6aa96e55b1fa49c767d562ba3ffe47 Mon Sep 17 00:00:00 2001
From: Kiran Sanjeevan <ksanjeevancabeza@gmail.com>
Date: Mon, 30 Jul 2018 00:00:09 -0700
Subject: [PATCH] Update README

---
 README.md | 4 ++--
 yolov2.py | 3 +--
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index d6e5675..cb5c964 100644
--- a/README.md
+++ b/README.md
@@ -206,7 +206,7 @@ Will store the custom bounding box priors wherever the path indicates in the con
 
 Training will create directory **logs/** which will store metrics and checkpoints for all the different training runs.
  
-Model passed is used for [transfer learning](https://en.wikipedia.org/wiki/Transfer_learning).
+Model passed is used for [transfer learning](https://en.wikipedia.org/wiki/Transfer_learning) by randomizing the last layer of the network (with the appropiate size of the target classes).
 
 Example:
 `python3 dourflow.py train -m models/logo/coco_model.h5 -c confs/config_custom.json`
@@ -222,7 +222,7 @@ Then, in another terminal tab you can run `tensorboard --logdir=logs/run_X` and
 
 #### To Do
 
-- [x] Multiclass NMS
+- [x] Multiclass Non Max Suppression
 - [x] Anchor generation for custom datasets
 - [ ] mAP write up
 - [x] Add webcam support
diff --git a/yolov2.py b/yolov2.py
index 7e122b1..32ce080 100644
--- a/yolov2.py
+++ b/yolov2.py
@@ -205,7 +205,7 @@ def training(self):
     
         early_stop = EarlyStopping(monitor='val_loss', 
                                min_delta=0.001, 
-                               patience=7, 
+                               patience=3, 
                                mode='min', 
                                verbose=1)
 
@@ -243,7 +243,6 @@ def training(self):
                         decay=0.0)
         
 
-
         map_cbck = Callback_MAP(generator=valid_gen, 
                                 model=self.inf_model, 
                                 tensorboard=tensorboard)