- 找到
/mmdet/visualization/local_visualizer.py
这个文件,从156行开始
for i, (pos, label) in enumerate(zip(positions, labels)):
# 这里先把类名拼接在了label_text中
if 'label_names' in instances:
label_text = instances.label_names[i]
else:
label_text = classes[
label] if classes is not None else f'class {label}'
if 'scores' in instances:
score = round(float(instances.scores[i]) * 100, 1)
# 这里再把分数拼接到label_text
#label_text += f': {score}'
# 所以想要不显示类名的话,这里可以采用这种写法,直接把分数的值赋给label_text即可(而非拼接),这样就只显示分数,这里自己怎么改都行
label_text = f'|{score}'
- 效果如下: