1.打开YOLOv5的源代码文件夹。
2.导航到 utils文件夹,找到general.py文件。
3.在该文件中,你会看到一个名为nms的函数,这是用于执行非最大抑制的函数。
4.在nms函数中,你可以找到一个变量conf_thres,它代表了置信度的阈值。默认情况下,它被设置为0.25.
5.你可以根据你的需求修改这个阈值。将其调高会过滤掉置信度较低的检测结果,而将其调低会保留更多的检测结果。
6.修改完阈值后,保存文件并重新运行YOLOv5,你将看到修改后的结果。
06-06
3万+
04-26
2万+
06-25