o 1jH"@sRddlZddlmZddlmZddlmZGdddeZded e fd d Z dS) N) BaseRunnergather_opts_args)CompletedProcessc @seZdZdZ ddedededededef d d Zdededededef d d Zdedededefd dZ ddZ dedededefddZ dedededefddZ dedededededef ddZ ddZdS) SegRunnerzSemantic Segmentation RunnerT config_pathcli_argsdeviceipssave_dirreturnc Csh|j|||d\}}||}g|d} |r| d| d|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 --do_eval--configTFenv switch_wdirechosilentcapture_outputlog_path) distributed_gather_opts_argsappendextendrun_cmd_get_train_log_path) selfrr r r r Zdo_evalargsrcmdr!k/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/repo_apis/PaddleSeg_api/seg/runner.pytrains   zSegRunner.trainc Csf|||\}}||}g|dd||}|j||ddddd}|jdkr1t|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/val.pyrTFrrrrrr)rrr returncode_extract_eval_metricsstdoutZmetrics) rrr r r rrr cp metric_dictr!r!r"evaluate@s    zSegRunner.evaluatecC.||}|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/predict.pyrTFrrrrpythonrrrr r r r!r!r"predictZs zSegRunner.predictc CsL|||\}}||}g|dd||}|j||ddddd}|S)analyseztools/analyse.pyrTFr$)rrr) rrr r r rrr r(r!r!r"r1ls  zSegRunner.analysecCs4||}|jddd|g|}|j|dddd}|S)a$run exporting Args: config_path (str): the path of config file used to export. cli_args (list): the additional parameters. device (str): unused. Returns: CompletedProcess: the result of exporting subprocess execution. ztools/export.py--for_fdrTFr,r-)rrr r r r(r!r!r"exportws  zSegRunner.exportcCr+)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. zdeploy/python/infer.pyrTFr,r-r/r!r!r"infers zSegRunner.infertrain_cli_argsexport_cli_argstrain_save_dirc Cs|j||d\}}||}g|ddd||}|j||dddd||d} ||}tj|dd } |jd d d|d | g|}|j|dddd } | | 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. rzdeploy/slim/quant/qat_train.pyrrTFrZ best_modelzmodel.pdparamszdeploy/slim/quant/qat_export.pyr2z --model_pathr,)rrrrospathjoinr.) rrr5r6r r7rrr Zcp_trainZ weight_pathZ cp_exportr!r!r" compressionsL   zSegRunner.compressioncCs t|dS)Nz--optsr)rrr!r!r"rs zSegRunner._gather_opts_argsN)T)__name__ __module__ __qualname____doc__strlistrr#r*r0r1r3r4r;rr!r!r!r"rsj  &    s   U