ObjectDetection-OneStageDet icon indicating copy to clipboard operation
ObjectDetection-OneStageDet copied to clipboard

当我的数据集只有一个类别时,模型不收敛

Open zhangming8 opened this issue 5 years ago • 3 comments

当我的数据集只有一个类别时,模型不收敛?请问有遇到这个问题吗

zhangming8 avatar Apr 02 '19 11:04 zhangming8

when you modify the class_num to 1, you should modify the loss function. Because Cross Entropy is not working on binary classification.

Jason-cs18 avatar May 28 '19 15:05 Jason-cs18

我也是只有一个类别,在控制台上打印class的概率总是0

shenghsiaowong avatar Jun 09 '19 11:06 shenghsiaowong

当我的数据集只有一个类别时,模型不收敛?请问有遇到这个问题吗

do you have solved it?

shenghsiaowong avatar Jul 01 '19 10:07 shenghsiaowong