Advertisement

notes on mask r-cnn code and reading integrated with the paper

阅读量:

mask r-cnn

  • 代码运行

    • 配置环境
    • 数据集
    • 跑train_net.py
    • 出现问题及解决方法
  • 代码学习

    • 打开工具包中的训练脚本tools/train_net.py
      • 搭建基于generalized_rcnn.py的模型架构
        • 首先构建backbone网络:build_backbone(backbone.py)

        • 接着添加FPN模块:fpn.py

        • build_rpn

          • RPN
        • RPNModule
        • RPNModule forward
        • RpnModule _forward_train
  • 在rpn.py中定义的RPNHead类用于处理相关操作。

  • 在anchor_generator.py文件中创建的生成器函数负责生成所需的 anchors。

  • 在AnchorGenerator类被初始化的过程中进行了必要的参数设置。

    • 具体来说,该类通过调用generate_anchors()方法来生成 anchors。

    • 在进行AnchorGenerator向前传播操作时,系统会自动处理数据流的传递

全部评论 (0)

还没有任何评论哟~