
    |-j6                     N    d dl Z d dlmZ ddlmZ ddlmZ  G d de          ZdS )	    N)Path   )BaseTrainer   )MODELSc                   4    e Zd ZdZeZdefdZd Zde	fdZ
dS )VideoDetTrainerz"Image Classification Model Trainersrc_label_dict_pathc                     t          | j        j                                      d          }t	          j        ||           dS )zzdump 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/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/modules/video_detection/trainer.pydump_label_dictzVideoDetTrainer.dump_label_dict   s@     #4#5#<==FFWW+-@AAAAA    c                 p   | j         j        r$| j                            | j         j                   | j         j        r$| j                            | j         j                   | j         j        r$| j                            | j         j                   | j         j        $| j        	                    | j         j                   | j        
                    | j        j        d           | j         j        dk    r$| j                            | j         j                   t          | j        j                                      d          }|                                r| j                            |           | j         j        $| j                            | j         j                   | j         j        $| j                            | j         j                   | j         j        $| j                            | j         j                   | j        j        &| j                            | j        j                   dS dS )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_configzVideoDetTrainer.update_config$   s   ) 	PO//0A0NOOO* 	RO001B1PQQQ* 	RO001B1PQQQ(4O..t/@/LMMM&&*,=	
 	
 	
 1R77O55!6   t1=>>GGXX!!## 	?O--o>>>'3O--d.?.JKKK*6O001B1PQQQ)5O**4+<+IJJJ$0O..t/A/HIIIII 10r   returnc                     d|                                  i}| j        j        | j        j        dk    r| j        j        |d<   | j                            dd          |d<   | j                            dd          |d<   |S )	zget key-value arguments of model training function

        Returns:
            dict: the arguments of training function.
        deviceNr   resume_pathdy2stFampOFF)
get_devicer   r6   get)r   
train_argss     r   get_train_kwargsz VideoDetTrainer.get_train_kwargsC   s      1 12
)5!-33(,(9(EJ}%"/33GUCC
7 -11%??
5r   N)__name__
__module____qualname____doc__r   entitiesstrr   r2   dictr=    r   r   r	   r	      sl        ,,HB3 B B B BJ J J>$      r   r	   )r   pathlibr   baser   
model_listr   r	   rE   r   r   <module>rI      sz                      < < < < <k < < < < <r   