
    9i6                     F    S SK r S SKJr  SSKJr  SSKJr   " S S\5      rg)	    N)Path   )BaseTrainer   )MODELSc                   >    \ rS rSrSr\rS\4S jrS r	S\
4S jrSrg	)
VideoDetTrainer   z"Image Classification Model Trainersrc_label_dict_pathc                     [        U R                  R                  5      R                  S5      n[        R
                  " X5        g)zbdump label dict config

Args:
    src_label_dict_path (str): path to label dict file to be saved.
label_map.txtN)r   global_configoutputjoinpathshutilcopyfile)selfr   dst_label_dict_paths      g/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/paddlex/modules/video_detection/trainer.pydump_label_dictVideoDetTrainer.dump_label_dict   s3     #4#5#5#<#<=FFW+A    c                 ,   U R                   R                  (       a/  U R                  R                  U R                   R                  5        U R                   R                  (       a/  U R                  R                  U R                   R                  5        U R                   R                  (       a/  U R                  R                  U R                   R                  5        U R                   R                  b/  U R                  R                  U R                   R                  5        U R                  R                  U R                  R                  S5        U R                   R                  S:w  a/  U R                  R                  U R                   R                  5        [        U R                  R                  5      R!                  S5      nUR#                  5       (       a  U R                  R%                  U5        U R                   R&                  b/  U R                  R)                  U R                   R&                  5        U R                   R*                  b/  U R                  R-                  U R                   R*                  5        U R                   R.                  b/  U R                  R1                  U R                   R.                  5        U R                  R2                  b0  U R                  R5                  U R                  R2                  5        gg)zupdate training configNVideoDetDataset r   )train_configlog_interval
pdx_configupdate_log_intervaleval_intervalupdate_eval_intervalsave_intervalupdate_save_intervalnum_classesupdate_num_classesupdate_datasetr   dataset_dirpretrain_weight_pathupdate_pretrained_weightsr   r   existsupdate_label_list
batch_sizeupdate_batch_sizelearning_rateupdate_learning_rateepochs_iters_update_epochsr   _update_output_dir)r   label_dict_paths     r   update_configVideoDetTrainer.update_config$   s   ))OO//0A0A0N0NO**OO001B1B1P1PQ**OO001B1B1P1PQ((4OO..t/@/@/L/LM&&**,=	
 11R7OO55!!66 t11==>GGX!!##OO--o>''3OO--d.?.?.J.JK**6OO001B1B1P1PQ))5OO**4+<+<+I+IJ$$0OO..t/A/A/H/HI 1r   returnc                 :   SU R                  5       0nU R                  R                  b3  U R                  R                  S:w  a  U R                  R                  US'   U R                  R                  SS5      US'   U R                  R                  SS5      US'   U$ )zkget key-value arguments of model training function

Returns:
    dict: the arguments of training function.
devicer   resume_pathdy2stFampOFF)
get_devicer   r9   get)r   
train_argss     r   get_train_kwargs VideoDetTrainer.get_train_kwargsC   s      12
))5!!--3(,(9(9(E(EJ}%"//33GUC
7 --11%?
5r    N)__name__
__module____qualname____firstlineno____doc__r   entitiesstrr   r4   dictr@   __static_attributes__rB   r   r   r	   r	      s*    ,HB3 BJ>$ r   r	   )r   pathlibr   baser   
model_listr   r	   rB   r   r   <module>rO      s       <k <r   