PyTorch_YOWO icon indicating copy to clipboard operation
PyTorch_YOWO copied to clipboard

你好,感谢分享demo。训练单类行为数据时,loss_cls一直未0正常吗。推理视频时检不出行为。

Open Eatzhy opened this issue 2 years ago • 1 comments

Eatzhy avatar Dec 12 '22 05:12 Eatzhy

@Eatzhy loss_cls=0是正常的,因为只有一个类别,那么softmax输出就是1,所以loss肯定是0的,所以,只有一个类别的时候,其实我们可以把cls预测注释掉,因为它没什么用,当然保留也是没问题的,反正loss都是0。检测不出行为这个问题原因可能有很多,一种是训练不够充分,一种可能是后处理需要做些相应调整?这个我就不确定了,得具体问题具体分析。

yjh0410 avatar Dec 12 '22 06:12 yjh0410