def METHOD_NAME(self, x):
    if self.training:
        pass
    else:
        x = self.backbone(x)
        x = self.neck(x)
        cls_scores, bbox_preds = self.head(x)
        prediction = torch.cat(
            [bbox_preds, cls_scores[..., 0:self.num_classes]], dim=-1)
        return prediction