YOLOv3_PyTorch icon indicating copy to clipboard operation
YOLOv3_PyTorch copied to clipboard

About the loss computation

Open chrisway613 opened this issue 4 years ago • 1 comments

I was confused about why use BCE to compute the loss of 'x' & 'y', could it be MSE?

https://github.com/BobLiu20/YOLOv3_PyTorch/blob/c6b483743598b5f64d520d81e7e5f47ba936d4c9/nets/yolo_loss.py#L55

chrisway613 avatar Apr 11 '20 03:04 chrisway613

I was confused about why use BCE to compute the loss of 'x' & 'y', could it be MSE?

https://github.com/BobLiu20/YOLOv3_PyTorch/blob/c6b483743598b5f64d520d81e7e5f47ba936d4c9/nets/yolo_loss.py#L55

I have same question with you

BigDataHa avatar Jul 16 '20 02:07 BigDataHa