mmdet框架理解

11 篇文章 0 订阅
6 篇文章 0 订阅

1.核心代码在mmdet/model目录下,每一个算法的最上层配置在detectors目录,里面会指定每一个算法的模型、训练、测试过程;在这些明确以后,就是配置backbones、neck、dense_heads。

2.通常loss的定义在head网络中指定。在backbones、neck、dense_heads的配置中,主要均包含两部分,网络结构和前向传播。为了模块解耦,把完整算法的网络分拆在不同模块中,把完整的forward过程分拆在不同模块中。但核心是,理解你看到的任何模板,核心都是包含这两部分,然后自己把他们穿起来就是完整的算法。

3.基于1和2,以mmdet框架新写一个算法的方法为:detectors目录下创建检测器的类型,依次创建对应需要的backbone、neck、head。

4.由于框架中的频繁嵌套调用,在传参数时,常用的格式是tuple和dict。即时返回值仅有一个Tensor,但是可能也会被再次打包为tuple,dict,格式错误容易导致奇怪的报错。

对于tensor类的两个变量a、b,

常见tuple类型返回为return a,b         

常见dict类型返回为 return dict{a=a,b=b} 

对于一个tensor变量a,

常见tuple类型返回为return a,               注意:加上逗号返回类型才是tuple         

常见dict类型返回为 return dict{a=a} 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值