o 1j@sFddlZddlmZddlmZGdddeZdedefd d ZdS) N) BaseRunner)CompletedProcessc @seZdZdZ ddedededededef d d Zdededededef d d Zdedededefd dZ ddededededef ddZ dedededefddZ dedededededef ddZ dS) ClsRunnerz Cls RunnerT config_pathcli_argsdeviceipssave_dirreturnc Cs^|j|||d\}}g|dd||} | dd|g|j| |dddd||dS) a&train model Args: config_path (str): the config file path used to train. cli_args (list): the additional parameters. device (str): the training device. ips (str): the ip addresses of nodes when using distribution. save_dir (str): the directory path to save training output. do_eval (bool, optional): whether or not to evaluate model during training. Defaults to True. Returns: CompletedProcess: the result of training subprocess execution. )Zlog_dirztools/train.py-c-ozGlobal.eval_during_train=TF)env switch_wdirechosilentcapture_outputlog_path) distributedextendrun_cmdZ_get_train_log_path) selfrrrr r Zdo_evalargsrcmdrl/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/repo_apis/PaddleClas_api/cls/runner.pytrainszClsRunner.trainc Cs\|||\}}g|dd||}|j||ddddd}|jdkr,t|j} | |_|S)arun model evaluating Args: config_path (str): the config file path used to evaluate. cli_args (list): the additional parameters. device (str): the evaluating device. ips (str): the ip addresses of nodes when using distribution. Returns: CompletedProcess: the result of evaluating subprocess execution. z tools/eval.pyr TF)rrrrrr)rr returncode_extract_eval_metricsstdoutZmetrics) rrrrr rrrcp metric_dictrrrevaluate;s   zClsRunner.evaluatecCs$|jdd|g|}|j|ddddS)a7run predicting using dynamic mode Args: config_path (str): the config file path used to predict. cli_args (list): the additional parameters. device (str): unused. Returns: CompletedProcess: the result of predicting subprocess execution. ztools/infer.pyr TFrrrpythonrrrrrrrrrpredictSszClsRunner.predictNcCs0|jdd|g|dd}|j|dddd}|S)arun exporting Args: config_path (str): the path of config file used to export. cli_args (list): the additional parameters. device (str): unused. save_dir (str, optional): the directory path to save exporting output. Defaults to None. Returns: CompletedProcess: the result of exporting subprocess execution. ztools/export_model.pyr r zGlobal.export_for_fd=TrueTFr#r$)rrrrr rr rrrexportds zClsRunner.exportcCs$|jdd|g|}|j|ddddS)a<run predicting using inference model Args: config_path (str): the path of config file used to predict. cli_args (list): the additional parameters. device (str): unused. Returns: CompletedProcess: the result of inferring subprocess execution. zpython/predict_cls.pyr ZdeployTFr#r$r&rrrinfers zClsRunner.infertrain_cli_argsexport_cli_argstrain_save_dirc CsN||||d|}tj|dd}g|dd|}||||}||fS)arun compression model Args: config_path (str): the path of config file used to predict. train_cli_args (list): the additional training parameters. export_cli_args (list): the additional exporting parameters. device (str): the running device. train_save_dir (str): the directory path to save output. Returns: CompletedProcess: the result of compression subprocess execution. NZ best_modelmodelr zGlobal.pretrained_model=)rospathjoinr() rrr*r+rr,Zcp_trainZ weight_pathZ cp_exportrrr compressionszClsRunner.compression)T)N) __name__ __module__ __qualname____doc__strlistrrr"r'r(r)r1rrrrrs|  #   rrr c Csddl}d}dd|dd|dd|dd|d d|g}d gd d ggd d gdgg}t}t||D]*\}}||}|D]}||} | r`t|tt| D]\} } | || <qWqDq7|S)zextract evaluation metrics from training log Args: stdout (str): the training log Returns: dict: the training metric rNz([-+]?[0-9]*\.?[0-9]+(?:[eE][-+]?[0-9]+)?z'\[Eval\]\[Epoch 0\]\[Avg\].*top1: (_dp)Z_dpz4\[Eval\]\[Epoch 0\]\[Avg\].*top1: (_dp), top5: (_dp)zF\[Eval\]\[Epoch 0\]\[Avg\].*recall1: (_dp), recall5: (_dp), mAP: (_dp)z<\[Eval\]\[Epoch 0\]\[Avg\].*MultiLabelMAP\(integral\): (_dp)z/\[Eval\]\[Epoch 0\]\[Avg\].*evalres:\ ma: (_dp)zval.top1zval.top5)Zrecall1Zrecall5ZmAPZ MultiLabelMAPz evalres: ma) rereplacedictzipcompile splitlinessearchmapfloatgroups) rr8Z_DPpatternskeysr!patternkeylinematchkvrrrrs:         r) r.baserZbase.utils.subprocessrrr6r:rrrrrs