def METHOD_NAME(self, x, return_score=False):
    """ Forward pass with input x. """
    score_map_ctr, size_map, offset_map = self.get_score_map(x)
    if return_score:
        bbox, max_score = self.cal_bbox(
            score_map_ctr, size_map, offset_map, return_score=True)
        return score_map_ctr, bbox, size_map, offset_map, max_score
    else:
        bbox = self.cal_bbox(score_map_ctr, size_map, offset_map)
        return score_map_ctr, bbox, size_map, offset_map