Fix

ksanjeevan · Jul 8, 2018 · 60a4435 · 60a4435
1 parent 63449c4
commit 60a4435
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/net/netloss.py b/net/netloss.py
@@ -94,7 +94,7 @@ def obj_loss(self, y_true, y_pred):
 
         norm_conf = 1
         if self.norm:
-            norm_conf = K.sum(K.cast((indicator_obj + indicator_noobj)  > 0.0), np.float32)
+            norm_conf = K.sum(K.cast((indicator_obj + indicator_noobj)  > 0.0, np.float32))
 
         indicator_o = indicator_obj + indicator_noobj
         loss_obj = K.sum(K.square(b_o-b_o_pred) * indicator_o, axis=[1,2,3])