MASK_RCNN解读6-网络误差反向向传播总体流程

MASK_RCNN解读6-网络误差反向向传播总体流程

从train_net.py的def main():函数进入,读取参数、初始化日志后进入def train(cfg, local_rank, distributed):函数,进行模型加载。通过下列代码读取模型权重值:
checkpointer = DetectronCheckpointer(
cfg, model, optimizer, scheduler, output_dir, save_to_disk
)
extra_checkpoint_data = checkpointer.load(cfg.MODEL.WEIGHT)
arguments.update(extra_checkpoint_data)
读取完成后,我们可以看到,除了ResNet的网络结构外,,其他层的的tensor的requires_grad被标记为TRUE,这个字段表示该tensor是否需要在被反向误差传递,主要的列表如下:

1)FPN网络(Feature Pyramid Networks,含金字塔)模块:
model.backbone.fpn.fpn_inner1
model.backbone.fpn.fpn_inner2
model.backbone.fpn.fpn_inner3
model.backbone.fpn.fpn_inner4
model.backbone.fpn.fpn_layer1
model.backbone.fpn.fpn_layer2
model.backbone.fpn.fpn_layer3
model.backbone.fpn.fpn_layer4

2)rpn 网络:
Regio

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值