쬜i U d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
m Z ddl m
Z
dd lmZ dd
lmZmZmZmZmZmZmZmZmZmZ dd lm Z d d
l m Z d dl m Z m!Z!m"Z"m#Z# ejH e% Z&g dZ'er e Z(ee)e*e)dz e)dz f f e+d< n
eg d Z(e(jY D ]! \ Z-\ Z.Z/ e sdZ. e sdZ/e.e/fe(e-< # e e e( Z0de)fdZ1 d"de)ejd z de)ejd z dz de3de4e)e)f dz de3e)z dz de)dz de3fdZ5d Z6 e d G d d Z7d!d gZ8y)#zAutoImageProcessor class. N) OrderedDict)
TYPE_CHECKING )PreTrainedConfig) get_class_from_dynamic_moduleresolve_trust_remote_code)ImageProcessingMixin)BaseImageProcessorFast)
CONFIG_NAMEIMAGE_PROCESSOR_NAMEPROCESSOR_NAME cached_fileis_timm_config_dictis_timm_local_checkpointis_torchvision_availableis_vision_availableloggingsafe_load_json_file)requires )_LazyAutoMapping)CONFIG_MAPPING_NAMES
AutoConfigmodel_type_to_module_name!replace_list_option_in_docstrings)ChameleonImageProcessorFlavaImageProcessorIdefics3ImageProcessorSmolVLMImageProcessor IMAGE_PROCESSOR_MAPPING_NAMES))aimv2CLIPImageProcessorCLIPImageProcessorFast)aimv2_vision_modelr" )alignEfficientNetImageProcessor EfficientNetImageProcessorFast)altclipr" )aria)AriaImageProcessorN)
aya_visionGotOcr2ImageProcessorGotOcr2ImageProcessorFast)beitBeitImageProcessorBeitImageProcessorFast)bitBitImageProcessorBitImageProcessorFast)blipBlipImageProcessorBlipImageProcessorFast)zblip-2r: ) bridgetower)BridgeTowerImageProcessor BridgeTowerImageProcessorFast) chameleon)r ChameleonImageProcessorFast) chinese_clip)ChineseCLIPImageProcessor ChineseCLIPImageProcessorFast)clipr" )clipsegViTImageProcessorViTImageProcessorFast)cohere2_vision)N Cohere2VisionImageProcessorFast)colpaliSiglipImageProcessorSiglipImageProcessorFast)colqwen2Qwen2VLImageProcessorQwen2VLImageProcessorFast)conditional_detr) ConditionalDetrImageProcessor!ConditionalDetrImageProcessorFast)convnextConvNextImageProcessorConvNextImageProcessorFast)
convnextv2rX )cvtrX )zdata2vec-visionr2 ) deepseek_vl)DeepseekVLImageProcessor DeepseekVLImageProcessorFast)deepseek_vl_hybrid) DeepseekVLHybridImageProcessor"DeepseekVLHybridImageProcessorFast)deformable_detr DeformableDetrImageProcessor DeformableDetrImageProcessorFast)deit)DeiTImageProcessorDeiTImageProcessorFast)depth_anythingDPTImageProcessorDPTImageProcessorFast) depth_pro)DepthProImageProcessorDepthProImageProcessorFast)detrDetrImageProcessorDetrImageProcessorFast)dinatrG )dinov2r6 )
dinov3_vit)NDINOv3ViTImageProcessorFast)z
donut-swin)DonutImageProcessorDonutImageProcessorFast)dptrk )edgetamNSam2ImageProcessorFast)efficientloftr) EfficientLoFTRImageProcessor EfficientLoFTRImageProcessorFast) efficientnetr' )emu3)Emu3ImageProcessorN)eomtEomtImageProcessorEomtImageProcessorFast) eomt_dinov3r )ernie4_5_vl_moe) Ernie4_5_VL_MoeImageProcessor!Ernie4_5_VL_MoeImageProcessorFast)flava)r FlavaImageProcessorFast) florence2r" )focalnetr6 )fuyu)FuyuImageProcessorFuyuImageProcessorFast)gemma3Gemma3ImageProcessorGemma3ImageProcessorFast)gemma3nrM )gitr" )glm46v)Glm46VImageProcessorGlm46VImageProcessorFast)glm4v)Glm4vImageProcessorGlm4vImageProcessorFast) glm_image)GlmImageImageProcessorGlmImageImageProcessorFast)glpn)GLPNImageProcessorGLPNImageProcessorFast)got_ocr2r. )zgrounding-dinoGroundingDinoImageProcessor GroundingDinoImageProcessorFast)groupvitr" )hierar6 )idefics)IdeficsImageProcessorN)idefics2)Idefics2ImageProcessorIdefics2ImageProcessorFast)idefics3)r Idefics3ImageProcessorFast)ijeparG )imagegpt)ImageGPTImageProcessorImageGPTImageProcessorFast) instructblipr: )internvlr. )janus)JanusImageProcessorJanusImageProcessorFast)zkosmos-2r" )z
kosmos-2.5)Kosmos2_5ImageProcessorKosmos2_5ImageProcessorFast)
layoutlmv2)LayoutLMv2ImageProcessor LayoutLMv2ImageProcessorFast)
layoutlmv3LayoutLMv3ImageProcessor LayoutLMv3ImageProcessorFast) layoutxlm)r r )levit)LevitImageProcessorLevitImageProcessorFast)lfm2_vl)NLfm2VlImageProcessorFast) lightglue)LightGlueImageProcessorLightGlueImageProcessorFast) lighton_ocrPixtralImageProcessorPixtralImageProcessorFast)llama4)NLlama4ImageProcessorFast)llava)LlavaImageProcessorLlavaImageProcessorFast)
llava_nextLlavaNextImageProcessorLlavaNextImageProcessorFast)llava_next_videor )llava_onevision) LlavaOnevisionImageProcessor LlavaOnevisionImageProcessorFast)lw_detrrd ) mask2former)Mask2FormerImageProcessor Mask2FormerImageProcessorFast)
maskformer)MaskFormerImageProcessor MaskFormerImageProcessorFast)
metaclip_2r" )zmgp-strrG )mistral3r )mlcdr" )mllama)MllamaImageProcessorMllamaImageProcessorFast)zmm-grounding-dinor ) mobilenet_v1)MobileNetV1ImageProcessor MobileNetV1ImageProcessorFast) mobilenet_v2)MobileNetV2ImageProcessor MobileNetV2ImageProcessorFast) mobilevitMobileViTImageProcessorMobileViTImageProcessorFast) mobilevitv2r )nougat)NougatImageProcessorNougatImageProcessorFast)z omdet-turborr ) oneformer)OneFormerImageProcessorOneFormerImageProcessorFast)ovis2)Ovis2ImageProcessorOvis2ImageProcessorFast)owlv2)Owlv2ImageProcessorOwlv2ImageProcessorFast)owlvit)OwlViTImageProcessorOwlViTImageProcessorFast) paddleocr_vl)PaddleOCRVLImageProcessor PaddleOCRVLImageProcessorFast) paligemmarM ) perceiver)PerceiverImageProcessorPerceiverImageProcessorFast)
perception_lm)N PerceptionLMImageProcessorFast)phi4_multimodal)N Phi4MultimodalImageProcessorFast)
pix2struct)Pix2StructImageProcessor Pix2StructImageProcessorFast)pixior6 )pixtralr )
poolformer)PoolFormerImageProcessor PoolFormerImageProcessorFast)pp_doclayout_v3)N PPDocLayoutV3ImageProcessorFast)prompt_depth_anything)!PromptDepthAnythingImageProcessor%PromptDepthAnythingImageProcessorFast)pvtPvtImageProcessorPvtImageProcessorFast)pvt_v2r ) qwen2_5_omnirQ )
qwen2_5_vlrQ )qwen2_vlrQ )qwen3_5rQ ) qwen3_5_moerQ )qwen3_omni_moerQ )qwen3_vlrQ )regnetrX )resnetrX )rt_detr)RTDetrImageProcessorRTDetrImageProcessorFast)samSamImageProcessorSamImageProcessorFast)sam2r} )
sam2_videor} )sam3NSam3ImageProcessorFast) sam3_trackerr6 )sam3_tracker_videor6 )
sam3_videor6 )sam_hqr0 ) segformerSegformerImageProcessorSegformerImageProcessorFast)seggpt)SegGptImageProcessorN) shieldgemma2r )sigliprM )siglip2)Siglip2ImageProcessorSiglip2ImageProcessorFast)smolvlm)r SmolVLMImageProcessorFast) superglue)SuperGlueImageProcessorSuperGlueImageProcessorFast)
superpoint)SuperPointImageProcessor SuperPointImageProcessorFast) swiftformerrG )swinrG )swin2sr)Swin2SRImageProcessorSwin2SRImageProcessorFast)swinv2rG )t5gemma2r )t5gemma2_encoderr )ztable-transformerrr )textnet)TextNetImageProcessorTextNetImageProcessorFast) timesformerVideoMAEImageProcessorN) timm_wrapper)TimmWrapperImageProcessorN)trocrrG )tvp)TvpImageProcessorTvpImageProcessorFast)udopr )upernetr= )
video_llama_3)VideoLlama3ImageProcessor VideoLlama3ImageProcessorFast) video_llava)VideoLlavaImageProcessorN)videomaer[ )vilt)ViltImageProcessorViltImageProcessorFast)vipllavar" )vitrG )vit_maerG )vit_msnrG )vitmatte)VitMatteImageProcessorVitMatteImageProcessorFast)vitpose)VitPoseImageProcessorVitPoseImageProcessorFast)xclipr" )yolos)YolosImageProcessorYolosImageProcessorFast)zoedepth)ZoeDepthImageProcessorZoeDepthImageProcessorFast
class_namec | dk( rt S t j D ]< \ }}| |v s t | }t j
d| d } t
|| c S t j j D ] }|D ] }t
|dd | k( s|c c S ! t j
d }t || r t
|| S y # t $ r Y w xY w)Nr
.ztransformers.models__name__ transformers) r
r itemsr importlib
import_modulegetattrAttributeErrorIMAGE_PROCESSOR_MAPPING_extra_contentvalueshasattr)r module_name
extractorsmodule extractor main_modules j/mnt/e/genesis-system/.venv/lib/python3.12/site-packages/transformers/models/auto/image_processing_auto.py#get_image_processor_class_from_namer s --%%#@#F#F#H Z #3K@K ,,q
->@UVF
vz22
.<<CCE !
# !Iy*d3zA !! )).9K{J'{J// "
s C
CC pretrained_model_name_or_path cache_dirforce_downloadproxiestokenrevisionlocal_files_onlyc t | t ||||||dd
}t | t ||||||dd
} | s|st j d i S i }
|t | } d| v r| d }
|
|
t | }
|
S )a
Loads the image processor configuration from a pretrained model image processor configuration.
Args:
pretrained_model_name_or_path (`str` or `os.PathLike`):
This can be either:
- a string, the *model id* of a pretrained model configuration hosted inside a model repo on
huggingface.co.
- a path to a *directory* containing a configuration file saved using the
[`~ProcessorMixin.save_pretrained`] method, e.g., `./my_model_directory/`.
cache_dir (`str` or `os.PathLike`, *optional*):
Path to a directory in which a downloaded pretrained model configuration should be cached if the standard
cache should not be used.
force_download (`bool`, *optional*, defaults to `False`):
Whether or not to force to (re-)download the configuration files and override the cached versions if they
exist.
proxies (`dict[str, str]`, *optional*):
A dictionary of proxy servers to use by protocol or endpoint, e.g., `{'http': 'foo.bar:3128',
'http://hostname': 'foo.bar:4012'}.` The proxies are used on each request.
token (`str` or *bool*, *optional*):
The token to use as HTTP bearer authorization for remote files. If `True`, will use the token generated
when running `hf auth login` (stored in `~/.huggingface`).
revision (`str`, *optional*, defaults to `"main"`):
The specific model version to use. It can be a branch name, a tag name, or a commit id, since we use a
git-based system for storing models and other artifacts on huggingface.co, so `revision` can be any
identifier allowed by git.
local_files_only (`bool`, *optional*, defaults to `False`):
If `True`, will only try to load the image processor configuration from local files.
Passing `token=True` is required when you want to use a private model.
Returns:
`Dict`: The configuration of the image processor.
Examples:
```python
# Download configuration from huggingface.co and cache.
image_processor_config = get_image_processor_config("google-bert/bert-base-uncased")
# This model does not have a image processor config so the result will be an empty dict.
image_processor_config = get_image_processor_config("FacebookAI/xlm-roberta-base")
# Save a pretrained image processor locally and you can reload its config
from transformers import AutoImageProcessor
image_processor = AutoImageProcessor.from_pretrained("google/vit-base-patch16-224-in21k")
image_processor.save_pretrained("image-processor-test")
image_processor_config = get_image_processor_config("image-processor-test")
```F) filenamer r r r r r _raise_exceptions_for_gated_repo%_raise_exceptions_for_missing_entriesz8Could not locate the image processor configuration file.image_processor)r r
r loggerinfor ) r r r r r r r kwargsresolved_processor_file resolved_image_processor_fileimage_processor_dictprocessor_dicts r get_image_processor_configr s D *% %))..3 %0%%%))..3 %! )1H NO
*,-DE .#12C#D $05I5Q 23P Q c 6 t j d| d y )NzFast image processor class zz is available for this model. Using slow image processor class. To use the fast image processor class set `use_fast=True`.)r warning)
fast_classs r '_warning_fast_image_processor_availabler s!
NN
%j\ 2g gr )vision)backendsc N e Zd ZdZd Ze ee d Ze dd Z
y)AutoImageProcessora%
This is a generic image processor class that will be instantiated as one of the image processor classes of the
library when created with the [`AutoImageProcessor.from_pretrained`] class method.
This class cannot be instantiated directly using `__init__()` (throws an error).
c t d )NzAutoImageProcessor is designed to be instantiated using the `AutoImageProcessor.from_pretrained(pretrained_model_name_or_path)` method.)OSError)selfs r __init__zAutoImageProcessor.__init__ s
d
r c |j dd }|j dd }|j dd }d|d< d|v r|j d }nt | rt }nt } t j
|fd|i|\ }} |j dd } d} d |j d
i v r|d
d } | V| T|j d d }
|
|
j d d
} d|j d
i v r|d
d }|j d d
} | `| ^t |t st j |fd|i|}t |dd } t |d
r d |j v r|j d } d}| |@| j# d }|s-t% r#| t& vrt( j+ d| d d}|r| j# d s| dz
} |r@t% s6t- | dd }|t/ d| d t( j+ d d}|r5t- | }| | dd } d}t( j+ d t- | }n>| j1 d }t- | }| | j# d rt/ d| d | du}|duxs t3 | t4 v }|rR| t | t6 s| df} |r | d | d }n| d }d|v r|j9 d d }nd}t; ||||| }|r\|rZ|s| d t= | d t? |fi |}|j d d } |jA |j |g|i |S | |j |g|i |S t3 | t4 v ret4 t3 | }|\ }}|s
| t= | |r|s| |j |g|i |S | |j |g|i |S t/ d t/ d | d t d t d!t d"d#jC d$ tD D
# t $ rH}
t j
|fdt i|\ }} n# t $ r |
w xY wt | s|
Y d}
~
d}
~
ww xY w)%a
Instantiate one of the image processor classes of the library from a pretrained model vocabulary.
The image processor class to instantiate is selected based on the `model_type` property of the config object
(either passed as an argument or loaded from `pretrained_model_name_or_path` if possible), or when it's
missing, by falling back to using pattern matching on `pretrained_model_name_or_path`:
List options
Params:
pretrained_model_name_or_path (`str` or `os.PathLike`):
This can be either:
- a string, the *model id* of a pretrained image_processor hosted inside a model repo on
huggingface.co.
- a path to a *directory* containing a image processor file saved using the
[`~image_processing_utils.ImageProcessingMixin.save_pretrained`] method, e.g.,
`./my_model_directory/`.
- a path or url to a saved image processor JSON *file*, e.g.,
`./my_model_directory/preprocessor_config.json`.
cache_dir (`str` or `os.PathLike`, *optional*):
Path to a directory in which a downloaded pretrained model image processor should be cached if the
standard cache should not be used.
force_download (`bool`, *optional*, defaults to `False`):
Whether or not to force to (re-)download the image processor files and override the cached versions if
they exist.
proxies (`dict[str, str]`, *optional*):
A dictionary of proxy servers to use by protocol or endpoint, e.g., `{'http': 'foo.bar:3128',
'http://hostname': 'foo.bar:4012'}.` The proxies are used on each request.
token (`str` or *bool*, *optional*):
The token to use as HTTP bearer authorization for remote files. If `True`, will use the token generated
when running `hf auth login` (stored in `~/.huggingface`).
revision (`str`, *optional*, defaults to `"main"`):
The specific model version to use. It can be a branch name, a tag name, or a commit id, since we use a
git-based system for storing models and other artifacts on huggingface.co, so `revision` can be any
identifier allowed by git.
use_fast (`bool`, *optional*, defaults to `False`):
Use a fast torchvision-base image processor if it is supported for a given model.
If a fast image processor is not available for a given model, a normal numpy-based image processor
is returned instead.
return_unused_kwargs (`bool`, *optional*, defaults to `False`):
If `False`, then this function returns just the final image processor object. If `True`, then this
functions returns a `Tuple(image_processor, unused_kwargs)` where *unused_kwargs* is a dictionary
consisting of the key/value pairs whose keys are not image processor attributes: i.e., the part of
`kwargs` which has not been used to update `image_processor` and is otherwise ignored.
trust_remote_code (`bool`, *optional*, defaults to `False`):
Whether or not to allow for custom models defined on the Hub in their own modeling files. This option
should only be set to `True` for repositories you trust and in which you have read the code, as it will
execute code present on the Hub on your local machine.
image_processor_filename (`str`, *optional*, defaults to `"config.json"`):
The name of the file in the model directory to use for the image processor config.
kwargs (`dict[str, Any]`, *optional*):
The values in kwargs of any keys which are image processor attributes will be used to override the
loaded values. Behavior concerning key/value pairs whose keys are *not* image processor attributes is
controlled by the `return_unused_kwargs` keyword parameter.
Passing `token=True` is required when you want to use a private model.
Examples:
```python
>>> from transformers import AutoImageProcessor
>>> # Download image processor from huggingface.co and cache.
>>> image_processor = AutoImageProcessor.from_pretrained("google/vit-base-patch16-224-in21k")
>>> # If image processor files are in a directory (e.g. image processor was saved using *save_pretrained('./test/saved_model/')*)
>>> # image_processor = AutoImageProcessor.from_pretrained("./test/saved_model/")
```configNuse_fasttrust_remote_codeT
_from_autoimage_processor_filenameimage_processor_typer auto_mapfeature_extractor_typeFeatureExtractorImageProcessorAutoFeatureExtractorFastz The image processor of type `a ` is now loaded as a fast processor by default, even if the model checkpoint was saved with a slow processor. This is a breaking change and may produce slightly different outputs. To continue using the slow processor, instantiate this class with `use_fast=False`. `zU` requires `torchvision` to be installed. Please install `torchvision` and try again.zcUsing `use_fast=True` but `torchvision` is not available. Falling back to the slow image processor.Fzz`use_fast` is set to `True` but the image processor class does not have a fast version. Falling back to the slow version.z\` does not have a slow version. Please set `use_fast=True` when instantiating the processor.r r z--
code_revisionzZThis image processor cannot be instantiated. Please make sure you have `Pillow` installed.z Unrecognized image processor in z2. Should have a `image_processor_type` key in its z of z3, or one of the following `model_type` keys in its z: z, c 3 K | ] }| y w)N ).0cs r z5AutoImageProcessor.from_pretrained.. s @jq@js )#popr r r r get_image_processor_dict Exceptionr getreplace
isinstancer r from_pretrainedr r r endswithr DEFAULT_TO_SLOW_IMAGE_PROCESSORSr warning_oncer
ValueError removesuffixtyper tuplesplitr r r register_for_auto_classjoinr )clsr inputsr r r r r config_dict_initial_exceptionr image_processor_auto_mapfeature_extractor_classfeature_extractor_auto_mapimage_processor_classimage_processor_class_fastimage_processor_type_slowhas_remote_codehas_local_code class_ref
upstream_repoimage_processor_tupleimage_processor_class_pys r r z"AutoImageProcessor.from_pretrained s X Hd+::j$/ "JJ':D A #| & /'-zz2L'M $
%&C
D'2 $'; $ ( 1 J J- H` dj NK( +/Et L #' ;??:r#B B'2:'>?S'T $ ',D,L&1oo6NPT&U #&2'>'F'FGY[k'l$% R)HH-8-DE[-\*+E+M+MN`br+s( ',D,L f&67#331&7 $+63I4#P vz*/Cv/V+1??;O+P( $ + /88@ 02,4TT''78L7M Nu u $H 4 = =f E$.$ 8 :(KL`adbdLe(f%(0$ 01 2G H ##y !-PQe-f*-5+?+D( $H''= )LL`(a%,@,M,Mf,U)(KLe(f%(05I5R5RSY5Z$ 01 2N O 3$>.d:ed6lNe>e '3JG_af;
mn n 2AA A+B<+P (M=)1-:*)1-:* (( 57QR]e )
r )NNF) r
__module__ __qualname____doc__r classmethodr r r staticmethodr r r r r r sF
&'DEl
F l
\
r r r )NFNNNF)9r r os collectionsr typingr configuration_utilsr dynamic_module_utilsr r image_processing_utilsr image_processing_utils_fastr
utilsr r r
r r r r r r r utils.import_utilsr auto_factoryr configuration_autor r r r
get_loggerr r r r strr __annotations__r
model_type
slow_classr r r PathLikebooldictr r r __all__r r r r s # 4 \ : A + *
H %$ VaUb!;sE#*cDj:P4Q/Q#Rb$/e
g%!T -J,O,O,Q I(J(Z
# %
1;Z0H!*-
I ++?A^_ C < +/ %) # "l #l R[[ 4'l l #s(^d
" l
#: l Dj
l l ^
; Z
Z
Z
z %&:
;r