search

bubbliiiing / yolov4-tiny-pytorch

这是一个YoloV4-tiny-pytorch的源码，可以用于训练自己的模型。
MIT License
787 stars 184 forks source link

关于分类loss #45

Closed Kaiwen6 closed 3 years ago

Kaiwen6 commented 3 years ago

泡泡哥好，我打算只对行人做检测，看了一篇论文用yolov3的，里面对loss部分做了修改，将分类loss直接删了，其他不变。我试了一下，最后出来的结果是行人的置信度普遍偏低相较于不删分类loss的，我就想问做单目标检测时需要删分类loss吗？bceloss怎么解释（此时只有背景和目标两类吗还是别的啥）

bubbliiiing commented 3 years ago

不需要。。

bubbliiiing commented 3 years ago

别想的那么复杂