本文介绍了如何在mmdetection框架下修改config文件来训练自己的目标检测模型。首先从Supported methods选择模型并获取config文件，复制到本地并修改，如调整num_classes、optimizer和data路径。针对num_classes，需在模型定义的roi_head bbox_head中设置。同时要注意，某些参数可能需要根据报错信息补充。修改__base__引用以匹配原有COCO配置，完成配置文件定制。
简单来说，首页open-mmlab/mmdetection: OpenMMLab Detection Toolbox and Benchmark (github.com)
Supported methods模块点连接，如faster rcnn
进入后选择模型，config为config 文件所在地址，复制到自己的config路径后修改，如./configs/fire
下载model放置./checkpoints
找到上级文件，按照格式覆盖上级所含的参数
最低0.47元/天 解锁文章
扫一扫
评论 2
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
