Prompt-Depth-Anything-Small / processing_prompt_depth_anything.py
bigshanedogg's picture
Upload Prompt-Depth-Anything-Small (transformers port, original code)
29ebc1d verified
Raw History Blame Contribute Delete
585 Bytes
"""Prompt Depth Anything processor: ``PromptDepthAnythingImageProcessor`` (the official load_image / load_depth) as its
only part."""
from transformers.processing_utils import ProcessorMixin
class PromptDepthAnythingProcessor(ProcessorMixin):
attributes = ["image_processor"]
def __init__(self, image_processor=None, **kwargs):
super().__init__(image_processor, **kwargs)
def __call__(self, images=None, prompt_depth=None, **kwargs):
return self.image_processor(images, prompt_depth=prompt_depth, **kwargs)
__all__ = ["PromptDepthAnythingProcessor"]