
    }-j                     j    d dl Z ddlmZmZmZ ddlmZmZ  G d de          Z G d d	e          Z	dS )
    N   )add_simple_inference_argsget_subcommand_argsperform_simple_inference   )PaddleXPredictorWrapperPredictorCLISubcommandExecutorc                   *     e Zd Zdd fd
Zd Z xZS )ImageClassificationN)topkc                L    d|i| _          t                      j        di | d S )Nr    )_extra_init_argssuper__init__)selfr   kwargs	__class__s      g/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddleocr/_models/_image_classification.pyr   zImageClassification.__init__   s9     D!
 	""6"""""    c                     | j         S N)r   r   s    r   &_get_extra_paddlex_predictor_init_argsz:ImageClassification._get_extra_paddlex_predictor_init_args%   s    $$r   )__name__
__module____qualname__r   r   __classcell__)r   s   @r   r   r      sV         	# 	# 	# 	# 	# 	# 	#% % % % % % %r   r   c                   J    e Zd Zd Zeej        d                         Zd ZdS )%ImageClassificationSubcommandExecutorc                 ^    t          |           |                    dt          d           d S )Nz--topkz#Top-k value for prediction results.)typehelp)r   add_argumentint)r   	subparsers     r   _update_subparserz7ImageClassificationSubcommandExecutor._update_subparser*   s@    !),,,6 	 	
 	
 	
 	
 	
r   c                     t           r   )NotImplementedErrorr   s    r   wrapper_clsz1ImageClassificationSubcommandExecutor.wrapper_cls3   s
     "!r   c                 N    t          |          }t          | j        |           d S r   )r   r   r*   )r   argsparamss      r   execute_with_argsz7ImageClassificationSubcommandExecutor.execute_with_args8   s(    $T** !16:::::r   N)	r   r   r   r'   propertyabcabstractmethodr*   r.   r   r   r   r    r    )   s[        
 
 
 " "  X"; ; ; ; ;r   r    )
r0   
_utils.clir   r   r   baser   r	   r   r    r   r   r   <module>r4      s    


         
 J I I I I I I I% % % % %1 % % % ; ; ; ; ;,J ; ; ; ; ;r   