资讯 小学 初中 高中 语言 会计职称 学历提升 法考 计算机考试 医护考试 建工考试 教育百科
栏目分类:
子分类:
返回
空麓网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
空麓网 > 计算机考试 > 软件开发 > 后端开发 > Python

【MMdetection训练及使用脚本系列】MMdetection训练1——如何保存最优的checkpoint文件

Python 更新时间: 发布时间: 计算机考试归档 最新发布

【MMdetection训练及使用脚本系列】MMdetection训练1——如何保存最优的checkpoint文件

MMdetection如何保存最优的checkpoint文件

以目标检测为例,进入到

configs/_base_/datasets/coco_detection.py

将evaluation = dict(interval=1, metric='bbox')改为evaluation = dict(interval=1, metric='bbox', save_best='auto')即可。

但是不建议这样做,防止以后忘记这次更改的数据。建议可以从自己每次单独训练的.py文件进行更改。

先来看用训练好的checkpoint文件跑test.py脚本的输出结果

最下方会输出一个有序字典OrderedDict,我们可以指定相关的键来保存最好的模型。

需要的操作就是在配置文件中进行如下修改:
例如,训练路径下atss_r50_fpn_1x_coco.py文件中的evaluation = dict(interval=1, metric='bbox') 改为evaluation = dict(interval=1, metric='bbox', save_best='auto')即可。

需要注意的是,save_best用于指定对应的键,'auto'是指保留第一个键对应最大值的checkpoint文件,即'bbox_mAP'(对应coco评价指标的第一行),也可以指定save_best='bbox_mAP_50'(coco评价指标的第二行)保留最大checkpoint文件。

保存的checkpoint文件路径在work_dir中,命名格式如下:

这样在训练过程中会自动保存自己指定指标最高的checkpoint文件。

转载请注明:文章转载自 http://www.konglu.com/
本文地址:http://www.konglu.com/it/1098154.html
免责声明:

我们致力于保护作者版权,注重分享,被刊用文章【【MMdetection训练及使用脚本系列】MMdetection训练1——如何保存最优的checkpoint文件】因无法核实真实出处,未能及时与作者取得联系,或有版权异议的,请联系管理员,我们会立即处理,本文部分文字与图片资源来自于网络,转载此文是出于传递更多信息之目的,若有来源标注错误或侵犯了您的合法权益,请立即通知我们,情况属实,我们会第一时间予以删除,并同时向您表示歉意,谢谢!

我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2023 成都空麓科技有限公司

ICP备案号:蜀ICP备2023000828号-2