MMDetection 是一个基于 PyTorch 的目标检测开源工具箱 ,支持了众多主流的和最新的检测算法,例如 Faster R-CNN,Mask R-CNN,RetinaNet 等,官网也给出了详细的教程。
训练好一个模型之后,我们接下来可能会考虑在新的图片上进行推理,看一下具体的效果,但是如果目标比较多,显示标注框、标签和置信度,会让图片显得异常拥挤,我们想做的是去掉显示结果框的文字标注以及置信度,那么该如何去实现呢?
路径:
/home/wangzhenkuan/mmdetection/mmdet/visualization/local_visualizer.py
本文介绍了如何在MMDetection中,于推理阶段移除目标检测结果上的文字标注和置信度显示。通过修改可视化函数,删除相关代码段,可以实现干净的图像展示。示例命令展示了不带文字和置信度的推理结果。
订阅专栏 解锁全文
6万+

被折叠的 条评论
为什么被折叠?



