You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

咨询:单类别签名检测场景下Facebook DETR-resnet-50的num_labels参数设置

关于DETR模型num_labels参数的设置说明
  • 直接将num_labels设为1即可,不需要额外添加“无签名”类别。
  • 原因:DETR模型内部会自动处理背景类(即无目标的区域),你只需要指定实际要检测的目标类别数量——这里就是签名这1个类别。模型的检测头会自动在输出中包含背景类的预测,无需手动在num_labels中额外计数。

对应代码修改为:

model = DetrForObjectDetection.from_pretrained(pretrained_model_name_or_path=CHECKPOINT, num_labels=1, ignore_mismatched_sizes=True)

内容的提问来源于stack exchange,提问作者Yusif Mukhtarov

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.06.26 01:28:10