mmdect之config

本文档详细记录了在MMDetection框架中如何配置模型以保存最佳模型,解决保存模型占用内存过多的问题。通过查阅文档和源码,作者找到了修改evaluation参数的方法,包括设置interval、save_best和allowed_metrics,最终成功实现按需保存模型。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >


我的环境 版本信息:
在这里插入图片描述

配置文件就在configs里面找
主要包括: model,data,optimizer等的设置
总之一切参数都可以在这里面设置。

保存最好的模型

例如想要保存最好的模型时:我用的mmdect是 默认 保存最后一个模型,并且每个epoch都会保存一次,很占用内存。
怎么样保存最好的模型呢?
首先google了一下,找到一个答案:https://github.com/open-mmlab/mmclassification/issues/125
截图如下:
在这里插入图片描述
意思是修改 evalutation config的这个evalutation参数;
但是哪里有这个参数呢?找到 configs/mask_rcnn/mask_rcnn_r50_caffe_c4_1x_coco.py 文件,发现只有这几个字典(字典就是配置参数)。
在这里插入图片描述

接着又再train中 debug了一下,发现读取的cfg文件里有默认的 work_flow,checkpoint_config , evalutation 等,也就是说没有显示的指定,就会采用这里的默认值。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值