
    9i                     R    S SK r S SKJr  SSKJr  SSKJr  SSKJr   " S	 S
\5      r	g)    N)Path   )abspath   )
ClsTrainer   )MODELSc                   (    \ rS rSrSr\rS rS rSr	g)FaceRecTrainer   zFace Recognition Model Trainerc                 d   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                  5         U R                   R                  b/  U R                  R                  U R                   R                  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%                  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                  b/  U R                  R5                  U R                   R2                  5        U R                  R6                  b0  U R                  R9                  U R                  R6                  5        gg)zupdate training configN 	label.txt)train_configlog_interval
pdx_configupdate_log_intervaleval_intervalupdate_eval_intervalsave_intervalupdate_save_intervalupdate_dataset_cfgnum_classesupdate_num_classespretrain_weight_pathupdate_pretrained_weightsr   global_configdataset_dirjoinpathexistsdump_label_dict
batch_sizeupdate_batch_sizelearning_rateupdate_learning_rateepochs_iters_update_epochswarmup_stepsupdate_warmup_epochsoutput_update_output_dir)selflabel_dict_paths     h/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/paddlex/modules/face_recognition/trainer.pyupdate_configFaceRecTrainer.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T!!##  1''3OO--d.?.?.J.JK**6OO001B1B1P1PQ))5OO**4+<+<+I+IJ))5OO001B1B1O1OP$$0OO..t/A/A/H/HI 1    c                    [        [        R                  R                  U R                  R
                  S5      5      n[        [        R                  R                  U R                  R
                  S5      5      n[        [        R                  R                  US5      5      n[        [        R                  R                  US5      5      nSSU 3SU 3SS	U 3S
U 3/nU R                  R                  U5        g )Ntrainvalr   zpair_label.txtz(DataLoader.Train.dataset.name=ClsDatasetz$DataLoader.Train.dataset.image_root=z(DataLoader.Train.dataset.cls_label_path=z,DataLoader.Eval.dataset.name=FaceEvalDatasetz%DataLoader.Eval.dataset.dataset_root=z(DataLoader.Eval.dataset.pair_label_path=)r   ospathjoinr   r   r   update)r,   train_dataset_dirval_dataset_dirtrain_list_pathval_list_pathds_cfgs         r.   r   !FaceRecTrainer.update_dataset_cfg;   s    #GGLL++77A
 ""'',,t/A/A/M/Mu"UV!"'',,/@+"NO_>N OP 723D2EF66GH:3O3DE6}oF
 	v&r1    N)
__name__
__module____qualname____firstlineno____doc__r	   entitiesr/   r   __static_attributes__r?   r1   r.   r   r      s    (HJ>'r1   r   )
r5   pathlibr   
utils.miscr   image_classificationr   
model_listr	   r   r?   r1   r.   <module>rK      s"    
  ! - 4'Z 4'r1   