From 2b7ada2c5c819808a174444e82622315a07fa11e Mon Sep 17 00:00:00 2001
From: Jacob Gildenblat <jacob.gildenblat@gmail.com>
Date: Wed, 2 Nov 2016 13:16:26 +0200
Subject: [PATCH] revert fix

---
 cam.py   | 2 +-
 model.py | 5 ++---
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/cam.py b/cam.py
index 35ea0b1..29982c4 100644
--- a/cam.py
+++ b/cam.py
@@ -7,7 +7,7 @@
 import argparse
 
 def train(dataset_path):
-        model = get_model(load_weights = False)
+        model = get_model()
         X, y = load_inria_person(dataset_path)
 	print "Training.."
         checkpoint_path="weights.{epoch:02d}-{val_loss:.2f}.hdf5"
diff --git a/model.py b/model.py
index 9419eb3..42df8d4 100644
--- a/model.py
+++ b/model.py
@@ -49,11 +49,10 @@ def VGG16_convolutions():
     model.add(Convolution2D(512, 3, 3, activation='relu', name='conv5_3'))
     return model
 
-def get_model(load_weights = True):
+def get_model():
     model = VGG16_convolutions()
 
-    if load_weights:
-        model = load_model_weights(model, "vgg16_weights.h5")
+    model = load_model_weights(model, "vgg16_weights.h5")
     
     model.add(Lambda(global_average_pooling, 
               output_shape=global_average_pooling_shape))