From 60a44356cb5fcad6456f8e56975ff44a17032c96 Mon Sep 17 00:00:00 2001 From: Kiran Sanjeevan Date: Sun, 8 Jul 2018 09:51:30 -0700 Subject: [PATCH] Fix --- net/netloss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/net/netloss.py b/net/netloss.py index b5b426e..448a90a 100644 --- 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])