咨询:单类别签名检测场景下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
相关产品推荐
相关产品推荐

