How to get max val bbox in all epoch?

[vansin]

背景 Background

类似 wandb 等工具只能记录某次训练的曲线，而不能自动的记录训练的最大值。
Tools like wandb can only record the curve of a certain training session, but not the maximum value of the training automatically.

本 Discussion 将通过阅读源码，给wandb添加此功能。
This Discussion will add this feature to wandb by reading the source code.

[vansin]

在日志中获取我们需要的信息

在 mmengine/runner/loops.py#L346 评估结果。

mmengine/hooks/runtime_info_hook.py#L101

并添加到 message_hub 里面

我们可以在 runner.message_hub.log_scalars['val/coco/bbox_mAP_50']._log_history 中获取到历史的所有数据，可以看到 log_scalars 字典中对应的是一个 numpy 数组。

由此我们遍可以在 runner.message_hub.log_scalars['val/coco/bbox_mAP_50']._log_history.max() 中获取最大值，由此我们可以进行下一步可视化工作了。

可视化相关记录

于是在mmyolo/engine/hooks/max_value_hook.py 添加了 MaxValueHook，来计算验证结果中的最高分值。