MMDetection 是一个基于 PyTorch 的目标检测开源工具箱 ,支持了众多主流的和最新的检测算法,例如 Faster R-CNN,Mask R-CNN,RetinaNet 等,官网也给出了详细的教程。
训练好一个模型之后,我们接下来可能会考虑在新的图片上进行推理,看一下具体的效果,但是如果目标比较多,显示标注框、标签和置信度,会让图片显得异常拥挤,我们想做的是去掉显示结果框的文字标注以及置信度,那么该如何去实现呢?
路径:
/home/wangzhenkuan/mmdetection/mmdet/visualization/local_visualizer.py
我们首先找到函数:DetLocalVisualizer(Visualizer):
这个函数是可视化的,我们只需要在这里面去掉对应的文字标注以及置信度即可!
接下来找到_draw_instances
函数&#