Method decode

mmrotate/core/bbox/coder/angle_coder.py:99–114

Circular Smooth Label Decoder. Args: angle_preds (Tensor): The csl encoding of angle offset for each scale level. Has shape (num_anchors * H * W, coding_len) Returns: list[Tensor]: Angle offset for each scale level.

(self, angle_preds)

        return smooth_label.scatter(1, radius_range, smooth_value)

    def decode(self, angle_preds):
        """Circular Smooth Label Decoder.

        Args:
            angle_preds (Tensor): The csl encoding of angle offset
                for each scale level.
                Has shape (num_anchors * H * W, coding_len)

        Returns:
            list[Tensor]: Angle offset for each scale level.
                Has shape (num_anchors * H * W, 1)
        """
        angle_cls_inds = torch.argmax(angle_preds, dim=1)
        angle_pred = ((angle_cls_inds + 0.5) *
                      self.omega) % self.angle_range - self.angle_offset
        return angle_pred * (math.pi / 180)