训练过程中 class loss并没有参与训练的疑问

deepcam-cn / yolov5-face

YOLO5Face: Why Reinventing a Face Detector (https://arxiv.org/abs/2105.12931) ECCV Workshops 2022)

GNU General Public License v3.0

2.11k stars 501 forks source link

Open yukaizhou opened 2 years ago

yukaizhou commented 2 years ago

非常感谢您的开源工作。当我用wider_face数据集训练时，类别是 1，此时class，即output[:,15]是没有参与训练的，那您开源的模型yolov5l-face.pt中output[:,15]是参与预测的，这样不会有问题吗？还是我理解的有问题，期待您的回复。感谢！

gatzf commented 2 years ago

兄弟，请问您试过调整类别数吗，我在修改类别数以后，程序卡到backward了

sunjian2015 commented 1 year ago

兄弟，请问您试过调整类别数吗，我在修改类别数以后，程序卡到backward了

你好，请问，怎么修改类别数啊？