mmdetection 修改config训练自己的detector

本文介绍了如何在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

找到上级文件,按照格式覆盖上级所含的参数

常用的需要改的参数为:

mmdetectionconfig文件用于定义模型的结构和超参数。根据引用,mmdetectionconfig文件位于mmdetection/configs目录下,每个模型都有一个对应的config文件。配置文件使用Python语法编写,并包含了各种参数和模块的定义。 引用提到了mmdetection/mmdet/apis目录,其中的inference.py文件中包含了init_detector和inference_detector两个函数。init_detector函数用于从配置文件和模型权重文件中初始化模型,inference_detector函数用于进行推理。这些API函数可以帮助用户快速加载和使用模型。 根据引用,mmdetection的官方GitHub地址是https://github.com/open-mmlab/mmdetection。用户可以在该地址找到模型的权重文件。mmdetection官方提供了model_zoo.md文件,其中列出了各个模型的权重下载链接。 为了使用mmdetection进行测试,用户需要导入mmcv和相应的API函数,并进行初始化和推理操作。引用中给出了一个测试代码的示例,其中使用了faster_rcnn_r50_fpn_1x_coco模型进行推理。 综上所述,mmdetectionconfig文件用于定义模型结构和超参数,API函数可以帮助用户加载和使用模型,官方GitHub地址提供了模型权重下载链接。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* *3* [mmdetection源码详细解读](https://blog.csdn.net/weixin_41560402/article/details/108169935)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 100%"] [ .reference_list ]
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值