Merge pull request chuanqi305#62 from lngao/patch-1

Update gen.py
jorgher · Apr 15, 2018 · 572fbf1 · 572fbf1
2 parents daef68a + 83d2ab5
commit 572fbf1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gen.py b/gen.py
@@ -364,12 +364,12 @@ def ssd_test(self):
     phase: TEST
   }
   detection_evaluate_param {
-    num_classes: 21
+    num_classes: %d
     background_label_id: 0
     overlap_threshold: 0.5
     evaluate_difficult_gt: false
   }
-}""" % (self.class_num, self.class_num))
+}""" % (self.class_num, self.class_num, self.class_num))
 
     def ssd_loss(self):
       print(