def get_rpn_cls_loss(self): rpn_cls_score = self.get_rpn_cls_score() rpn_labels = self.get_rpn_labels() return rpn_cls_loss(rpn_cls_score,rpn_labels)
def get_rpn_cls_loss(self): assert self.eval_mode is False, 'No RPN cls loss without ground truth boxes' rpn_cls_score = self.get_rpn_cls_score() rpn_labels = self.get_rpn_labels() return rpn_cls_loss(rpn_cls_score, rpn_labels)