search

GZQ0723 / YoloV4

object detection \ yolov4
50 stars 28 forks source link

nn.bce loss #3

Open henbucuoshanghai opened 4 years ago

henbucuoshanghai commented 4 years ago

loss_conf_obj = self.bce_loss(pred_conf[obj_mask], tconf[obj_mask]) loss_conf_noobj = self.bce_loss(pred_conf[noobj_mask], tconf[noobj_mask])

why using the nn.bce loss ? why not the softmax loss?