【计算机视觉 | MMDetection】mmdetection推理时,如何去掉显示结果框的文字标注以及置信度?

MMDetection 是一个基于 PyTorch 的目标检测开源工具箱 ,支持了众多主流的和最新的检测算法,例如 Faster R-CNN,Mask R-CNN,RetinaNet 等,官网也给出了详细的教程。

训练好一个模型之后,我们接下来可能会考虑在新的图片上进行推理,看一下具体的效果,但是如果目标比较多,显示标注框、标签和置信度,会让图片显得异常拥挤,我们想做的是去掉显示结果框的文字标注以及置信度,那么该如何去实现呢?

路径:

/home/wangzhenkuan/mmdetection/mmdet/visualization/local_visualizer.py

在这里插入图片描述
我们首先找到函数:DetLocalVisualizer(Visualizer):

这个函数是可视化的,我们只需要在这里面去掉对应的文字标注以及置信度即可!

接下来找到_draw_instances函数&#

猜你喜欢

转载自blog.csdn.net/wzk4869/article/details/135299520