def METHOD_NAME(self,
                 x: 'Tensor',
                 raw_out: bool = False,
                 out_shape: Optional[Tuple[int, int]] = None) -> BoxList:
    out: List[BoxList] = super().predict(x, raw_out=raw_out)
    out = self.postprocess_model_output(x, out, out_shape=out_shape)
    return out