Yolov5中使用torch.hub.load()调用模型时如何隐藏预测置信度？

直接解决方案 #

调用results.show()时，直接传入hide_conf=True参数即可，这是最直接有效的方式：

import torch

# 加载YOLO模型
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
# 执行推理
img = 'test.jpg'  # 支持图片路径、numpy数组等格式
results = model(img)
# 显示结果时隐藏置信度
results.show(hide_conf=True)

其他可行方式 #

如果需要多次调用show()，可以先修改Result对象的hide_conf属性，后续调用show()时会自动生效：

results.hide_conf = True
results.show()  # 此时已隐藏置信度

为什么之前的设置无效 #

你尝试的model_damage_annot.hide_conf=True不生效，是因为模型对象的hide_conf属性并不控制可视化时的置信度显示。YOLO的可视化配置（包括是否显示置信度）是由推理后生成的Result对象管理的，而非模型本身的属性。

扩展：保存图片时隐藏置信度 #

如果需要保存不带置信度的检测结果图，同样可以给save()方法传hide_conf=True参数：

results.save(hide_conf=True)

内容的提问来源于stack exchange，提问作者Chirag Patankar