
    9i                     :    S SK r SSKJr  SSKJr   " S S\5      rg)    N   )BaseTrainer   )MODELSc                   0    \ rS rSrSr\rS rS\4S jr	Sr
g)
SegTrainer   z#Semantic Segmentation Model Trainerc                    U R                   R                  U R                  R                  S5        U R                  R
                  b/  U R                   R                  U R                  R
                  5        U R                  R                  (       aJ  U R                  R                  S:w  a/  U R                   R                  U R                  R                  SS9  ggg)zupdate training config
SegDatasetN T)is_backbone)	
pdx_configupdate_datasetglobal_configdataset_dirtrain_confignum_classesupdate_num_classespretrain_weight_pathupdate_pretrained_weights)selfs    m/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/paddlex/modules/semantic_segmentation/trainer.pyupdate_configSegTrainer.update_config   s    &&t'9'9'E'E|T((4OO..t/@/@/L/LM22!!66"<OO55!!66D 6  = 3    returnc                 >   SU R                  5       0n[        R                  R                  SS5        U R                  R
                  b  U R                  R
                  US'   U R                  R                  b  U R                  R                  US'   U R                  R                  b  U R                  R                  US'   U R                  R                  b3  U R                  R                  S:w  a  U R                  R                  US'   U R                  R                  b  U R                  R                  US	'   U R                  R                  (       a  U R                  R                  US
'   U R                  R                  (       a  SUS'   U R                  R                  US'   U R                  R                  SS5      US'   U R                  R                  S5      b  U R                  R                  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.
deviceFLAGS_npu_jit_compileN
batch_sizelearning_rateepochs_itersr   resume_pathsave_dir	log_itersTdo_evalsave_intervaldy2stFinput_shapeampOFF)
get_deviceosenvironpopr   r    r!   r"   r#   r   outputlog_intervaleval_intervalgetr)   )r   
train_argss     r   get_train_kwargsSegTrainer.get_train_kwargs(   s     12



.5''3'+'8'8'C'CJ|$**6*.*;*;*I*IJ'))5)-):):)G)GJ~&))5!!--3(,(9(9(E(EJ}%$$0%)%7%7%>%>Jz"))&*&7&7&D&DJ{#**$(Jy!*.*;*;*I*IJ'"//33GUC
7  /;(,(9(9(E(EJ}% --11%?
5r    N)__name__
__module____qualname____firstlineno____doc__r   entitiesr   dictr5   __static_attributes__r7   r   r   r   r      s    -H $  r   r   )r-   baser   
model_listr   r   r7   r   r   <module>rB      s     
  2 2r   