def METHOD_NAME(self, input: Dict[str, Any]) -> Dict[str, Any]:
    output_img = (input['output_tensor'].squeeze(0) * 255).cpu().permute(
        1, 2, 0).numpy().astype('uint8')
    return {OutputKeys.OUTPUT_IMG: output_img[:, :, ::-1]}