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])