你好，我想问下这个目标检测任务中有做目标分类吗？我看这个展示的结果都没有目标种类。尝试了改一下代码中的设置也没有成功，请大佬赐教！

[hddlovefxx]

[RuiwenZhang]

同问

[xinshoushangludd]

utils\plot.py 里面的函数 plot_one_box 那儿 将if label后面的 不要隐去 ，就行了

[Yaoxingtian]

pos_weight

你说的是这个吗？ BCEcls = nn.BCEWithLogitsLoss(pos_weight=torch.Tensor([cfg.LOSS.CLS_POS_WEIGHT])).to(device)，没有pos_weight=torch.ones([det_num_class])， 这里是权重的参数，没有det_num_class

[thinkthinking]

作者在github上的预训练模型包括代码默认都是只识别single class的，如果想要多类别识别，需要进行修改网络结构以及训练参数，重新训练。