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)
- 其中包含ResNet网络模块:resnet.py
-
接着添加FPN模块:fpn.py
-
build_rpn
-
- RPN
-
-
- RPNModule
- RPNModule forward
- RpnModule _forward_train
- 搭建基于generalized_rcnn.py的模型架构
- 打开工具包中的训练脚本tools/train_net.py
-
在rpn.py中定义的RPNHead类用于处理相关操作。
-
在anchor_generator.py文件中创建的生成器函数负责生成所需的 anchors。
-
在AnchorGenerator类被初始化的过程中进行了必要的参数设置。
-
具体来说,该类通过调用generate_anchors()方法来生成 anchors。
-
在进行AnchorGenerator向前传播操作时,系统会自动处理数据流的传递
-
全部评论 (0)
还没有任何评论哟~
