
    9i>                     n   S SK JrJr  S SKrS SKJr  S SKJr  S SK	J
r
Jr  S SKJr  S SKJr  S SKJr  \R$                  " \R&                  \R&                  S	9\R$                  " \R(                  \R*                  S	9\R$                  " \R(                  \R,                  S	9 " S
 S\5      5       5       5       rg)    )AnyDictN)	Pipelines)
OutputKeys)InputPipeline)	PIPELINES)	LoadImage)Tasks)module_namec                      ^  \ rS rSrS\4U 4S jjrS\S\\\4   4S jr	S\\\4   S\\\4   4S jr
S\\\4   S\\\4   4S	 jrS
rU =r$ )ImageDetectionPipeline   modelc                 ,   > [         TU ]  " SUSS.UD6  g)z$
model: model id on modelscope hub.
F)r   auto_collateN )super__init__)selfr   kwargs	__class__s      p/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/pipelines/cv/image_detection_pipeline.pyr   ImageDetectionPipeline.__init__   s     	Cu5CFC    inputreturnc                     [         R                  " U5      nUR                  [        R                  5      nU R
                  R                  U5      nSU0nU$ )Nimg)r
   convert_to_ndarrayastypenpfloat64r   
preprocess)r   r   r   results       r   r$   !ImageDetectionPipeline.preprocess   sF    **51jj$jj##C(r   c                 J    U R                   R                  US   5      nSU0nU$ )Nr   data)r   	inference)r   r   outputsr%   s       r   forwardImageDetectionPipeline.forward&   s)    **&&uU|4'"r   inputsc                    U R                   R                  US   5      u  p#nUc4  [        R                  / [        R                  / [        R
                  / 0nU$ [        R                  U[        R                  U[        R
                  U0nU$ )Nr(   )r   postprocessr   SCORESLABELSBOXES)r   r-   bboxesscoreslabelsr*   s         r   r/   "ImageDetectionPipeline.postprocess,   s    !%!7!7v!G>!!2!!2  "G
 Nvvf

 r   r   )__name__
__module____qualname____firstlineno__strr   r   r   r   r$   r+   r/   __static_attributes____classcell__)r   s   @r   r   r      sw    Dc D $sCx. T#s(^ S#X $sCx. T#s(^  r   r   )typingr   r   numpyr"   modelscope.metainfor   modelscope.outputsr   modelscope.pipelines.baser   r   modelscope.pipelines.builderr	   modelscope.preprocessorsr
   modelscope.utils.constantr   register_modulehuman_detectionimage_object_detectionobject_detectionabnormal_object_detectionr   r   r   r   <module>rK      s      ) ) 5 2 . + 	y'@'@B
	  i.H.HJ
	  335%X %5JB%r   