transformers/docs/source/en/main_classes/image_processor.md

1.3 KiB

Image Processor

An image processor is in charge of preparing input features for vision models and post processing their outputs. This includes transformations such as resizing, normalization, and conversion to PyTorch, TensorFlow, Flax and Numpy tensors. It may also include model specific post-processing such as converting logits to segmentation masks.

ImageProcessingMixin

autodoc image_processing_utils.ImageProcessingMixin - from_pretrained - save_pretrained

BatchFeature

autodoc BatchFeature

BaseImageProcessor

autodoc image_processing_utils.BaseImageProcessor