YOLOv5输出检测到的边界框的坐标、类别以及置信度
然后在plot_one_box函数中添加一行write_coordinates_to_file(c1, c2)即可,坐标会被写进一个coordinates文件。若想输出坐标和置信度,则在plot.py中的plot_one_box函数中,if lable:语句末尾加上print(lable)即可。在detect.py中找到plot_one_box,大概在113行,可按ctrl+F进行查找。ctrl+鼠标左键点击,会跳转到plot.py,并进入该函数。这样预测边界框的坐标、类别、置信度都有了直观的输出。
原创
2023-07-13 17:10:19 ·
8416 阅读 ·
14 评论