Trace-Anything / processing_trace_anything.py
bigshanedogg's picture
Upload Trace-Anything (transformers port, original code)
653a5eb verified
Raw History Blame Contribute Delete
499 Bytes
"""Trace Anything processor: ``TraceAnythingImageProcessor`` (infer.py _load_images) as its only part."""
from transformers.processing_utils import ProcessorMixin
class TraceAnythingProcessor(ProcessorMixin):
attributes = ["image_processor"]
def __init__(self, image_processor=None, **kwargs):
super().__init__(image_processor, **kwargs)
def __call__(self, images=None, **kwargs):
return self.image_processor(images, **kwargs)
__all__ = ["TraceAnythingProcessor"]