fix mrcnn_bbox_loss

keineahnung2345 · Jan 11, 2019 · 760ad47 · 760ad47
1 parent a0a2aa0
commit 760ad47
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mrcnn/model.py b/mrcnn/model.py
@@ -1137,7 +1137,7 @@ def mrcnn_bbox_loss_graph(target_bbox, target_class_ids, pred_bbox):
     loss = K.switch(tf.size(target_bbox) > 0,
                     smooth_l1_loss(y_true=target_bbox, y_pred=pred_bbox),
                     tf.constant(0.0))
-    loss = K.mean(loss)
+    loss = K.mean(K.sum(loss, axis=1))
     return loss