-
-
Notifications
You must be signed in to change notification settings - Fork 668
Open
Description
Hi, I think that axis=(1,2,3) should be specified in np.sum
here
| score_loss = tf.reduce_sum(score_loss * foreground_background_mask, axis=[1,2,3]) / np.sum(foreground_background_mask) |
and here
| boxes_loss = tf.reduce_sum(boxes_loss * foreground_mask, axis=[1,2,3]) / np.sum(foreground_mask) |
Metadata
Metadata
Assignees
Labels
No labels