yolo3-pytorch icon indicating copy to clipboard operation
yolo3-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

关于损失函数

Open TTgogogo opened this issue 3 years ago • 3 comments

损失函数都需要自定义吗？现在是坐标损失+类别损失+置信度损失，xy损失用BCE，wh用MSE，类别用BCE，置信度用BCE, 用BCE还是MSE是按照公式采用，还是随便选用的？

TTgogogo avatar Jun 28 '22 06:06 TTgogogo

随便用是什么意思？ 回归损失部分用的是giou

bubbliiiing avatar Jun 28 '22 14:06 bubbliiiing

用MSE还是BCE是根据公式得出的吗？

TTgogogo avatar Jun 29 '22 06:06 TTgogogo

不太理解你的意思，是根据训练目标选用

bubbliiiing avatar Jun 30 '22 15:06 bubbliiiing