yolov5训练过程中,怎么把进度条下显示的准确度召回率map等显示小数点后四位
时间: 2024-02-12 14:04:35 浏览: 97
在 YOLOv5 训练过程中,可以通过更改 `utils.general.metrics` 函数中的 `format` 参数来控制显示小数点后的位数。默认情况下,`format` 参数设置为 `%.3f`,即显示小数点后三位。若要显示小数点后四位,则可以将该参数更改为 `%.4f`。
具体操作如下:
1. 打开 `utils/general.py` 文件。
2. 找到 `metrics` 函数,该函数的代码如下:
```python
def metrics(pred, target, iou_thresh=0.5):
"""Calculate mean average precision (mAP), precision, recall for single image"""
...
...
...
```
3. 在该函数中,找到 `format` 参数,该参数默认设置为 `%.3f`。将其更改为 `%.4f`。
```python
def metrics(pred, target, iou_thresh=0.5):
"""Calculate mean average precision (mAP), precision, recall for single image"""
...
...
...
format_str = '%s' + '%10.4f' * 6 # 设置小数点后四位
print(format_str % ('all', *x))
```
4. 保存修改后的文件,重新启动训练即可看到小数点后四位的显示。
阅读全文