mmdetection中如何保存save best epoch的checkpoints;MMdetection如何保存训练过程中评价指标最好/最优的轮次的模型

mmdetection中如何保存save best epoch的checkpoints;MMdetection如何保存训练过程中评价指标最好/最优的轮次的模型_第1张图片api的训练文件中需要

eval_hook = DistEvalHook if distributed else EvalHook
runner.register_hook(
    eval_hook(val_dataloader, save_best='bbox_mAP', **eval_cfg), priority='LOW')

将evalhook通过register_hook注册进入钩子列表里,实例化时加入参数save_best=就可,需要什么具体评价指标,就赋什么指标名称 例如'bbox_mAP',类型必须为字符串。

注意:evalhook虽然命名为hook,但是没有使用HOOK.registry_module()进行注册表注册 所以也就不需要bulid。

如何保存save_best具体参见 DistEvalHook

mmdetection中如何保存save best epoch的checkpoints;MMdetection如何保存训练过程中评价指标最好/最优的轮次的模型_第2张图片

mmdetection中如何保存save best epoch的checkpoints;MMdetection如何保存训练过程中评价指标最好/最优的轮次的模型_第3张图片

 

 

你可能感兴趣的:(深度学习,tensorflow,人工智能)