训练二分类

[DONTKK]

作者你好，感谢你的开源代码！我目前在用你的代码跑实验，我的数据集仅包含一个类别，有含有前景物体的图像及其仅包含背景的图像，这样的话我在训练模型的时候该如何设置numclass呢，目前我设置了两个类别，背景图像的标签是0，含有前景的图像标签设置为1，这样操作会不会有什么影响，有没有好的建议？

[YuqiYang213]

您好，我们并没有试过在单分类问题上进行实验，但是从经验上来看，建议在含有前景的图象标签上设置为0和1，然后使用多标签分类的loss进行训练。除此之外，因为只有两个类，所以显著性标签可能也不需要了。
希望能帮助到您。