j+fSrSSKJr SSKrSSKJrJr SSKJr SSK J r J r J r J r JrJrJrJrJrJr SSKJr SSKJrJrJrJr SS KJr \ (aSS KJr \R>"\ 5r!"S S \"\5r#"S S\5r$SSjr%"SS\5r&"SS\5r'"SS\5r("SS\5r)g)z4Base classes for object-detection inference engines.) annotationsN)ABCabstractmethod)Enum) TYPE_CHECKINGAnyClassVarDictListLiteralOptionalTypeget_args get_origin)Image) BaseModel ConfigDictFieldfield_validator)PydanticUndefined)EngineModelConfigc8\rSrSrSrSrSrSr\S Sj5r Sr g) ObjectDetectionEngineType"z=Supported inference engine types for object-detection models. onnxruntime transformers api_kserve_v2c XR1;$)z0Check if an engine type is a remote API variant.) API_KSERVE_V2)cls engine_types s/home/mande/repo/quber/.venv/lib/python3.13/site-packages/docling/models/inference_engines/object_detection/base.pyis_remote_variant+ObjectDetectionEngineType.is_remote_variant)s00111N)r!rreturnbool) __name__ __module__ __qualname____firstlineno____doc__ ONNXRUNTIME TRANSFORMERSr classmethodr#__static_attributes__r&r%r"rr"s&GK!L#M22r%rc^\rSrSr%Sr\"SS9r\"SS9rS\ S'\"S S S 9r S \ S '0r S\ S'\ U4Sj5r SrU=r$) BaseObjectDetectionEngineOptions/z:Base configuration shared across object-detection engines.Tarbitrary_types_allowedzType of inference engine to use descriptionrr!g333333?z9Minimum confidence score to keep a detection (0.0 to 1.0))defaultr8floatscore_thresholdzQClassVar[dict[ObjectDetectionEngineType, Type[BaseObjectDetectionEngineOptions]]] _registryc p>[TU]"S0UD6 U[LagURR S5nU(dgSnUR n[ U5[La[U5n[U5S:XaUSnUcUR[La URnUbU[RU'gg)Nr!rr&) super__pydantic_init_subclass__r3 model_fieldsget annotationrr rlenr9rr<)r kwargsfieldr!annvalues __class__s r"r@;BaseObjectDetectionEngineOptions.__pydantic_init_subclass__As *4V4 2 2   $$]3   c?g %c]F6{a$Qi   5==8I#I--K  "FI , 6 6{ C #r%r&)r)r*r+r,r-r model_configrr!__annotations__r;r<r0r@r1 __classcell__)rIs@r"r3r3/shDd;L-25.K*#OOU  JJr%r3cSSKJn U"5$)Nr(TransformersObjectDetectionEngineOptions)1docling.datamodel.object_detection_engine_optionsrPrOs r"_default_engine_optionsrR_s 4 55r%cR\rSrSr%\"\SS9rS\S'\"SSS9\ S55r S r g ) !ObjectDetectionEngineOptionsMixiniz5Runtime configuration for the object-detection enginedefault_factoryr8r3engine_optionsbefore)modec[U[5(aU$[U[5(aHURS5n[RRU5nU(aUR U5$U$)Nr!) isinstancer3dictrBr<model_validate)r valuer! model_clss r"resolve_engine_options8ObjectDetectionEngineOptionsMixin.resolve_engine_optionsosc e= > >L eT " "))M2K8BBFF{SI //66 r%r&N) r)r*r+r,rrRrXrLrr0rar1r&r%r"rTrTis>7</K8N4 %H5 6 r%rTcT\rSrSr%Sr\"SS9r\"SS9rS\ S'\"\ S S 9r S \ S 'S r g)ObjectDetectionEngineInputz8Generic input accepted by every object-detection engine.Tr5zPIL image to run inference onr7rimagez3Optional metadata that is echoed back in the outputrVDict[str, Any]metadatar&N) r)r*r+r,r-rrKrrfrLr]rhr1r&r%r"rdrds6Bd;L%DEE5E$I Hnr%rdcz\rSrSr%Sr\"\SS9rS\S'\"\SS9r S\S '\"\S S9r S \S '\"\ S S9r S\S'Sr g)ObjectDetectionEngineOutputz,Output returned by object-detection engines.zPredicted class indicesrVz List[int] label_idsz%Confidence scores for the predictionsz List[float]scoresz8Bounding boxes as [x_min, y_min, x_max, y_max] in pixelszList[List[float]]bboxesz6Optional metadata echoed back from the input or enginergrhr&N)r)r*r+r,r-rlistrlrLrmrnr]rhr1r&r%r"rjrjsh6 -Iy ;FK!&N!F  %L Hnr%rjc\rSrSrSrS S Sjjr\S Sj5r\SSj5r\SSj5r SSjr SS jr S r g)BaseObjectDetectionEnginez1Abstract base-class for object-detection engines.Nc*XlX lSUlg)zInitialize the engine. Args: options: Engine-specific configuration options model_config: Model configuration (repo_id, revision, extra_config) FN)optionsrK _initialized)selfrtrKs r"__init__"BaseObjectDetectionEngine.__init__s (!r%cg)zBInitialize engine resources (load models, allocate buffers, etc.).Nr&rvs r" initialize$BaseObjectDetectionEngine.initializer%cg)z#Run inference on a batch of inputs.Nr&)rv input_batchs r" predict_batch'BaseObjectDetectionEngine.predict_batchr}r%cg)z`Get the label mapping for this model. Returns: Dictionary mapping label IDs to label names Nr&rzs r"get_label_mapping+BaseObjectDetectionEngine.get_label_mappingr}r%cUR(d9[RS[U5R5 UR 5 UR U/5nUS$)z*Helper to run inference on a single input.z%Initializing %s for single predictionr)ru_logdebugtyper)r{r)rv input_dataresultss r"predict!BaseObjectDetectionEngine.predictsL  JJ>T @S@S T OO $$j\2qzr%cUR(d9[RS[U5R5 UR 5 [ U[5(aURU5$URU5$)NzInitializing %s for call) rurrrr)r{r\rorr)rvrs r"__call__"BaseObjectDetectionEngine.__call__s]  JJ14:3F3F G OO  j$ ' '%%j1 1||J''r%)rurKrt)N)rtr3rKzOptional[EngineModelConfig]r'None)r'r)rz List[ObjectDetectionEngineInput]r'z!List[ObjectDetectionEngineOutput])r'zDict[int, str])rrdr'rj)rz=ObjectDetectionEngineInput | List[ObjectDetectionEngineInput]r'z?ObjectDetectionEngineOutput | List[ObjectDetectionEngineOutput]) r)r*r+r,r-rwrr{rrrrr1r&r%r"rqrqs; 59 "1 "2 "  "QQ2;2 *22    4  $  (Q ( I (r%rq)r'r3)*r- __future__rloggingabcrrenumrtypingrrr r r r r rrr PIL.Imagerpydanticrrrr pydantic_corer#docling.datamodel.stage_model_specsr getLoggerr)rstrrr3rRrTrdrjrqr&r%r"rs;"#   BB+E" 2T 2-Jy-J`6 .  )*9(9(r%