MMDetection3.0以上如何在推理是不显示类名?

  1. 找到/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}'
  1. 效果如下:
    MMDetection3.0以上如何在推理是不显示类名?_第1张图片

你可能感兴趣的:(python,mmdetection,pytorch)