MMDetection3.0以上如何在推理时不显示类名？

小勺子哦

已于 2023-12-06 15:52:37 修改

阅读量304

点赞数

文章标签： python mmdetection pytorch

于 2023-11-23 20:37:04 首次发布

本文链接：https://blog.csdn.net/qq_41544842/article/details/134585864

版权

文章介绍了如何在MMDet的`local_visualizer.py`文件中，从第156行起修改代码，实现仅显示对象的分数，而不显示类别名。作者提供了具体的操作方法和代码示例。

摘要由CSDN通过智能技术生成

找到/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}'