yolov5-6.0版本训练过程中,怎么把进度条下显示的准确度召回率map等显示小数点后四位,在哪个文件里,大概修改哪一行参数
时间: 2024-02-12 13:04:45 浏览: 131
yolov5-6.0-fire-smoke-1-火焰烟雾检测模型.zip
5星 · 资源好评率100%
Yolov5-6.0版本的训练过程中,在`train.py`文件的`train`函数中,可以找到如下代码:
```
# Print training results
loss_items = [x.item() for x in loss]
n = x.size(0)
```
这里的`loss_items`是一个包含训练中各项损失的列表。要将输出的准确度、召回率、mAP等显示小数点后四位,可以在这段代码后面添加以下代码:
```
# Print training results with four decimal points
metrics = []
for i, metric in enumerate(metric_items):
metrics.append(f"{metric}: {metrics[-1]:.4f}")
print(f"Epoch {epoch} {phase} metrics: {' | '.join(metrics)}")
```
这里的`metric_items`是一个包含各项评估指标的列表。这段代码会将每个评估指标的值输出到小数点后四位。
阅读全文