def METHOD_NAME(self, data: Tuple[Any, Any]) -> Dict[str, Any]:
    image, target = data  # They're both numpy arrays at this point
    return {
        "image": Image(image.transpose(2, 0, 1)),
        "label": Label(target.item(), categories=self._categories),
    }