examples/pytorch/semantic-segmentation/run_semantic_segmentation.py/ConvertImageDtype/__call__
class ConvertImageDtype:    def __call__(self, image, target):
        image = functional.convert_image_dtype(image, self.dtype)
        return image, target