def METHOD_NAME(self, batched_inputs):
    images = [x['image'].to(self.device) for x in batched_inputs]
    images = [self.normalizer(x) for x in images]
    images = ImageList.from_tensors(images,
                                    self.backbone.size_divisibility)
    return images