object detection yolov5中如何只显示得分大于0.80的boundary box?

How to show only boundary box with more than 0.80 score in object detection yolov5?

下面的检测图像,我只想显示0.8-0.99。 我尝试在 detect.py 上编辑 '''conf_thres=0.25'''、'''iou_thres=0.45''' 但没有任何更改输出。

请支持我。我如何只显示边界框 0.80 - 0.99 分数而小于 0.80 分数不显示?

很简单,使用detect.py

时需要将conf_thresh设为0.8

测试时使用标志--conf,然后在你的案例中输入你想要的数字0.8。 所以,你会得到这样的东西:

!python detect.py --weights models/best.pt --img 640 --conf 0.8 --source /path/to/test/images