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)