revert fix

jacobgil · Nov 2, 2016 · 2b7ada2 · 2b7ada2
1 parent 29eec01
commit 2b7ada2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git 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
@@ -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))