| """Camera Depth Model processor: ``CameraDepthModelImageProcessor`` (infer.py load_images + image2tensor) as its only part.""" | |
| from transformers.processing_utils import ProcessorMixin | |
| class CameraDepthModelProcessor(ProcessorMixin): | |
| attributes = ["image_processor"] | |
| def __init__(self, image_processor=None, **kwargs): | |
| super().__init__(image_processor, **kwargs) | |
| def __call__(self, images=None, depth=None, **kwargs): | |
| return self.image_processor(images, depth=depth, **kwargs) | |
| def post_process_depth_estimation(self, outputs, target_sizes=None): | |
| return self.image_processor.post_process_depth_estimation(outputs, target_sizes) | |
| __all__ = ["CameraDepthModelProcessor"] | |