port squeezeDet+, eval scripts to 1.0.0

elliothe · Mar 3, 2017 · f3b30e5 · f3b30e5
1 parent 969f47b
commit f3b30e5
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 5 deletions.
diff --git a/scripts/train.sh b/scripts/train.sh
@@ -62,9 +62,6 @@ then
   --gpu=$USE_GPU
 fi
 
-# =========================================================================== #
-# command for vgg16:
-# =========================================================================== #
 if [[ "$1" == "vgg16" ]]
 then
   python ./src/train.py \

diff --git a/src/eval.py b/src/eval.py
@@ -178,7 +178,7 @@ def evaluate():
 
     saver = tf.train.Saver(model.model_params)
 
-    summary_writer = tf.train.SummaryWriter(FLAGS.eval_dir, g)
+    summary_writer = tf.summary.FileWriter(FLAGS.eval_dir, g)
 
     ckpts = set() 
     while True:

diff --git a/src/nets/squeezeDetPlus.py b/src/nets/squeezeDetPlus.py
@@ -102,4 +102,4 @@ def _fire_layer(self, layer_name, inputs, s1x1, e1x1, e3x3, freeze=False):
         layer_name+'/expand3x3', sq1x1, filters=e3x3, size=3, stride=1,
         padding='SAME', freeze=freeze)
 
-    return tf.concat(3, [ex1x1, ex3x3], name=layer_name+'/concat')
+    return tf.concat([ex1x1, ex3x3], 3, name=layer_name+'/concat')