YOLOv3_TensorFlow2 icon indicating copy to clipboard operation
YOLOv3_TensorFlow2 copied to clipboard

confused with ignore_mask and true_object_mask

Open speeding-motor opened this issue 5 years ago • 1 comments

ignore_mask = best_ious < ignore_threshold
confidence_loss = true_object_mask * self.__binary_crossentropy_keep_dim(true_object_mask, pred_features[..., 4:5], from_logits=True) + \
                              (1 - true_object_mask) * self.__binary_crossentropy_keep_dim(true_object_mask, pred_features[..., 4:5], from_logits=True) * ignore_mask

when compute the confidence_loss, here you use the ignore_mask, and only in here, other place all use the true_object_mask, I m really confused with this part...

any reply is appreciate, thanks in advance @calmisential

speeding-motor avatar May 27 '20 15:05 speeding-motor

The role of ignore_mask is to ignore some negative samples when calculating the confidence loss.

calmiLovesAI avatar Jun 18 '20 02:06 calmiLovesAI