faster rcnn 源码（1）——GeneralizedRCNN
💡
原文中文，约4100字，阅读约需10分钟。
📝
内容提要
faster rcnn是继承于GeneralizedRCNN的模型，包含了transform、backbone、rpn和roi_heads四个重要接口。transform用于标准化和缩放图片，backbone提取特征，rpn生成proposals和proposal_losses，roi_heads进行roi pooling和分类。模型的前向传播函数接收images和targets作为输入，返回损失和检测结果。其中images是一个列表，每个元素是一个Tensor表示一张图片，targets是一个列表，每个元素是一个字典包含真实方框的坐标信息。模型的输出包含损失和检测成果，损失以字典形式返回，检测成果以列表形式返回，每个元素是一个字典包含检测框的置信度、类别标签和切割掩码等信息。模型还进行了图像的转换和后处理操作。
🎯
关键要点
- faster rcnn 继承于 GeneralizedRCNN 模型
- 模型包含 transform、backbone、rpn 和 roi_heads 四个重要接口
- transform 用于标准化和缩放图片
- backbone 提取特征，通常使用 VGG、ResNet、MobileNet 等网络
- rpn 生成 proposals 和 proposal_losses
- roi_heads 进行 roi pooling 和分类
- 前向传播函数接收 images 和 targets 作为输入
- images 是一个列表，每个元素是一个 Tensor 表示一张图片
- targets 是一个列表，每个元素是一个字典包含真实方框的坐标信息
- 模型输出包含损失和检测成果，损失以字典形式返回
- 检测成果以列表形式返回，每个元素是一个字典包含检测框的置信度、类别标签和切割掩码等信息
- transform 过程记录原始图像的尺寸以便后续处理
- 通过 backbone 提取特征后，使用 rpn 生成候选框和损失
- roi_heads 处理候选区域以生成最终检测结果
- 后处理阶段将检测框映射回原始图像尺寸
🏷️
标签
➡️
