开始使用免费开始使用

为 RPN 和 R-CNN 定义损失函数

您计划训练一个同时包含 RPN 和 R-CNN 组件的目标检测模型。要开始训练,您需要分别为每个组件定义其对应的损失函数。

请回忆:RPN 组件要判断一个候选区域内是否存在目标,并预测该候选区域的边界框坐标。R-CNN 组件需要将目标分类到多个类别之一,同时预测最终的边界框坐标。

已导入 torchtorch.nn 并命名为 nn

本练习是课程的一部分

使用 PyTorch 进行图像深度学习

查看课程

练习说明

  • 定义 RPN 的分类损失函数,并将其赋给 rpn_cls_criterion
  • 定义 RPN 的回归损失函数,并将其赋给 rpn_reg_criterion
  • 定义 R-CNN 的分类损失函数,并将其赋给 rcnn_cls_criterion
  • 定义 R-CNN 的回归损失函数,并将其赋给 rcnn_reg_criterion

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Implement the RPN classification loss function
rpn_cls_criterion = ____

# Implement the RPN regression loss function
rpn_reg_criterion = ____

# Implement the R-CNN classification Loss function
rcnn_cls_criterion = ____

# Implement the R-CNN regression loss function
rcnn_reg_criterion = ____
编辑并运行代码