image_processor

Utility classes for image processing.

Classes

BaseImageProcessor

Base class for image processors.

MllamaImageProcessor

Image processor for Mllama.

class BaseImageProcessor

Bases: object

Base class for image processors.

__init__(tokenizer, device='auto')

Constructor.

collate_function(examples)

Collate function to process images during data loading.

preprocess_function(examples)

Preprocess function.

class MllamaImageProcessor

Bases: BaseImageProcessor

Image processor for Mllama.

collate_function(batch)

Collate function to process images during data loading.

preprocess_function(examples)

Preprocess function.