o 1Îjr8ã@sXddlZddlmZddlmZddlmZddlmZddl m Z Gd d „d eƒZ dS) éNé)Úlogging)Úabspathé)Ú BaseModel)Ú CLIArgument)ÚCompletedProcessc@s6eZdZdZ           d$deded ed ed ed ed edededededefdd„Z     d%deded ed edededefdd„Z   d&dededed ededef dd„Z dededefdd„Z   d'deded edededef d d!„Z      d(dededed ed edededefd"d#„ZdS))ÚClsModelzImage Classification ModelNÚgpuFÚOFFTÚ batch_sizeÚ learning_rateÚ epochs_itersÚipsÚdeviceÚ resume_pathÚdy2stÚampÚ num_workersÚuse_vdlÚsave_dirÚreturnc  Ks¦|durt|ƒ}| ¡;} |j ¡}| |¡| |¡| | ¡|dur+| |¡|dur4| |¡|dur=|  |¡|  |¡| durKt| ƒ} nt|  ¡ƒ} |  | ¡| dur_|  | ¡g}|  dd¡}|  dd¡}|dury| td|ƒ¡|  dd¡}|durô| dd¡}| dd¡}| dd ¡}| d d¡} | |¡| | d d¡¡| | d d ¡¡| | d d ¡¡| | dd¡¡| | dd¡¡| durÖ|  | ¡|durß| |¡|duró| ¡D] \}}t|ƒtj|<qçn| |¡| d¡d}|  dd¡}|  dd ¡}| d|›g¡| d|j›g¡|r+| d|›g¡|  | ¡| !| ¡|j"j#| |||| |dWdƒS1sLwYdS)ahtrain self Args: batch_size (int, optional): the train batch size value. Defaults to None. learning_rate (float, optional): the train learning rate value. Defaults to None. epochs_iters (int, optional): the train epochs value. Defaults to None. ips (str, optional): the ip addresses of nodes when using distribution. Defaults to None. device (str, optional): the running device. Defaults to 'gpu'. resume_path (str, optional): the checkpoint file path to resume training. Train from scratch if it is set to None. Defaults to None. dy2st (bool, optional): Enable dynamic to static. Defaults to False. amp (str, optional): the amp settings. Defaults to 'OFF'. num_workers (int, optional): the workers number. Defaults to None. use_vdl (bool, optional): enable VisualDL. Defaults to True. save_dir (str, optional): the directory path to save train output. Defaults to None. Returns: CompletedProcess: the result of training subprocess execution. NÚdo_evalTÚprofilez--profiler_optionsÚ benchmarkÚenvÚseedFrrZdaliÚshuffleZ shared_memoryZprint_mem_infoú:rÚuniform_output_enabledÚexport_with_pirúGlobal.uniform_output_enabled=úGlobal.pdx_model_name=úGlobal.export_with_pir=)r)$rÚ_create_new_config_fileÚconfigÚcopyÚ update_deviceZ_update_to_staticÚ_update_use_vdlÚupdate_batch_sizeÚupdate_learning_rateÚ_update_epochsZ_update_checkpointsÚget_train_save_dirÚ_update_output_dirÚupdate_num_workersÚpopÚappendrÚgetZupdate_log_ranksÚ _update_ampZ update_daliZupdate_shuffleZupdate_shared_memoryZupdate_print_mem_infoZ update_seedÚitemsÚstrÚosÚenvironÚsplitÚupdateÚnameÚdumpÚ_assert_empty_kwargsÚrunnerÚtrain)Úselfr r rrrrrrrrrÚkwargsÚ config_pathr%Zcli_argsrrrZenvsrÚenv_nameZ env_valueZ device_typerr ©rBúk/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/repo_apis/PaddleClas_api/cls/model.pyr=sv"                       €      ÿ&ÂzClsModel.trainÚ weight_pathc Ksªt|ƒ}| ¡C}|j ¡} |  |¡|  |¡|  |¡|dur&|  |¡|dur/|  |¡|   |¡|  |¡|j   |g||¡} | WdƒS1sNwYdS)aševaluate self using specified weight Args: weight_path (str): the path of model weight file to be evaluated. batch_size (int, optional): the batch size value in evaluating. Defaults to None. ips (str, optional): the ip addresses of nodes when using distribution. Defaults to None. device (str, optional): the running device. Defaults to 'gpu'. amp (str, optional): the AMP setting. Defaults to 'OFF'. num_workers (int, optional): the workers number in evaluating. Defaults to None. Returns: CompletedProcess: the result of evaluating subprocess execution. N) rr$r%r&r2r'Úupdate_pretrained_weightsr)r.r:r;r<Úevaluate) r>rDr rrrrr?r@r%ÚcprBrBrCrF‚s         $ðzClsModel.evaluateÚ input_pathÚinput_list_pathc Ks t|ƒ}t|ƒ}|rt|ƒ}| ¡4}|j ¡}| |¡| ||¡| |¡| |¡| |¡|  |¡|j   |g|¡WdƒS1sIwYdS)aJpredict using specified weight Args: weight_path (str): the path of model weight file used to predict. input_path (str): the path of image file to be predicted. input_list_path (str, optional): the paths of images to be predicted if is not None. Defaults to None. device (str, optional): the running device. Defaults to 'gpu'. save_dir (str, optional): the directory path to save predict output. Defaults to None. Returns: CompletedProcess: the result of predicting subprocess execution. N) rr$r%r&rEZ_update_predict_imgr'Z_update_save_predict_resultr:r;r<Úpredict) r>rDrHrIrrr?r@r%rBrBrCrJ­s        $ôzClsModel.predictc Ksø| d¡s t|ƒ}t|ƒ}| ¡a}|j ¡}| |¡| |¡| dd¡}|r.| |¡| dd¡}| dd¡}|  d|›g¡|  d |j ›g¡|rX|  d |›g¡|  |¡|  |¡|j  |gd|¡WdƒS1suwYdS) a3export the dynamic model to static model Args: weight_path (str): the model weight file path that used to export. save_dir (str): the directory path to save export output. Returns: CompletedProcess: the result of exporting subprocess execution. ÚhttprNrTr Fr!r"r#)Ú startswithrr$r%r&rEZ_update_save_inference_dirr/r'r8r9r:r;r<Úexport) r>rDrr?r@r%rrr rBrBrCrMÕs(           $ìzClsModel.exportÚ model_dirÚ dict_pathc Ksät|ƒ}t|ƒ}|durt d¡tj |d¡}|j ¡}| |¡|  |¡|  |¡|  |¡|dur>t|ƒ}|  |¡d|vrJ|  | d¡¡| ¡}| |¡| |¡|j |g|¡WdƒS1skwYdS)a=predict image using infernece model Args: model_dir (str): the directory path of inference model files that would use to predict. input_path (str): the path of image that would be predict. device (str, optional): the running device. Defaults to 'gpu'. save_dir (str, optional): the directory path to save output. Defaults to None. dict_path (str, optional): the label dict file path. Defaults to None. Returns: CompletedProcess: the result of inferring subprocess execution. Nz`save_dir` will not be used.z inference.ymlZ enable_mkldnn)rrÚwarningr5ÚpathÚjoinr%r&ÚloadZ_update_inference_model_dirZ_update_infer_imgZ_update_infer_deviceZupdate_label_dict_pathZ_update_enable_mkldnnr/r$r:r;r<Úinfer) r>rNrHrrrOr?r@r%rBrBrCrTùs(          $ûzClsModel.inferc Ks t|ƒ}| ¡~} |j ¡} |  d¡|  |¡|  |¡|  |jd¡|   |¡|dur3|   |¡|dur<|   |¡|durE|   |¡|durNt|ƒ}nt|   ¡ƒ}|  |¡|  | ¡g} |  tddtj |d¡›ƒ¡| |¡|j | g| ||¡WdƒS1s‰wYdS)acompression model Args: weight_path (str): the path to weight file of model. batch_size (int, optional): the batch size value of compression training. Defaults to None. learning_rate (float, optional): the learning rate value of compression training. Defaults to None. epochs_iters (int, optional): the epochs or iters of compression training. Defaults to None. device (str, optional): the device to run compression training. Defaults to 'gpu'. use_vdl (bool, optional): whether or not to use VisualDL. Defaults to True. save_dir (str, optional): the directory to save output. Defaults to None. Returns: CompletedProcess: the result of compression subprocess execution. NZauto_compression_config_pathz-ozGlobal.save_inference_dir=rM)rr$r%r&r2r'r(Z_update_slim_configZ model_inforEr)r*r+r,r-r:r0rr5rQrRr;r<Ú compression) r>rDr r rrrrr?r@r%Zexport_cli_argsrBrBrCrU%s>             þÿ  ÿ$ßzClsModel.compression) NNNNr NFr NTN)NNr r N)Nr N)r NN)NNNr TN)Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚintÚfloatr4Úboolrr=rFrJrMrTrUrBrBrBrCr säôþýüûúùø ÷ ö õ ô òjùþýüûúù ÷/úþýüûú ø((úþýüûú ø/øþýüûúùø ör ) r5ÚutilsrZ utils.miscrÚbaserZbase.utils.argrZbase.utils.subprocessrr rBrBrBrCÚs