o 1jQS@sddlZddlZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z ddl mZmZddlmZdd lmZdd lmZmZmZd d gZdd dZeZGdd d ejdZGdddeZGddde Z!Gddde!Z"Gddde!Z#Gddde!Z$dS)N)flagslogging) get_cache_dir) parse_device)UnsupportedAPIErrorUnsupportedParamError)CachedProperty)Config)build_model_from_model_infobuild_runner_from_model_infoget_registered_model_info PaddleModel BaseModelc Cs|dur |dur td|dur|dur||jkrtdn |dur)|dur)|j}zt|}WntyE}z tt|d|d}~wwt||dS) _create_modelNz;At least one of `model_name` and `config` must be not None.zgIf both `model_name` and `config` are not None, `model_name` should be the same as `config.model_name`.z is not a registered model name.) model_infoconfig) ValueError model_namerKeyErrorrreprr )rrrer]/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/repo_apis/base/model.pyr&s*    rcs8eZdZdZdZdZd-fdd Zej         d.d d Z ej    d/d dZ ejd0ddZ ejddZ ejd0ddZ ej     d1ddZejddZejddZeddZeddZedd Zed!d"Zed#d$Zed%d&Zed'd(Zed)d*Zd+d,ZZS)2rz Abstract base class of Model. Model defines how Config and Runner interact with each other. In addition, Model provides users with multiple APIs to perform model training, prediction, etc. )trainevaluatepredictexportinfer compressionzsupported_{api_name}_optsNcsRt||_t||_t|j|_|dur tdt |}||_ | dS)z Initialize the instance. Args: model_name (str): A registered model name. config (base.config.BaseConfig|None): Config. Default: None. NaWe strongly discourage leaving `config` unset or setting it to None. Please note that when `config` is None, default settings will be used for every unspecified configuration item, which may lead to unexpected result. Please make sure that this is what you intend to do.) super__init__namerrr runnerrwarningr r _patch_apis)selfrr __class__rrr"Js    zBaseModel.__init__gpuFOFFTc Kt)a Train a model. Args: batch_size (int|None): Number of samples in each mini-batch. If multiple devices are used, this is the batch size on each device. If None, use a default setting. Default: None. learning_rate (float|None): Learning rate of model training. If None, use a default setting. Default: None. epochs_iters (int|None): Total epochs or iterations of model training. If None, use a default setting. Default: None. ips (str|None): If not None, enable multi-machine training mode. `ips` specifies Paddle cluster node ips, e.g., '192.168.0.16,192.168.0.17'. Default: None. device (str): A string that describes the device(s) to use, e.g., 'cpu', 'gpu', 'gpu:1,2'. Default: 'gpu'. resume_path (str|None): If not None, resume training from the model snapshot corresponding to the weight file `resume_path`. If None, use a default setting. Default: None. dy2st (bool): Whether to enable dynamic-to-static training. Default: False. amp (str): Optimization level to use in AMP training. Choices are ['O1', 'O2', 'OFF']. Default: 'OFF'. num_workers (int|None): Number of subprocesses to use for data loading. If None, use a default setting. Default: None. use_vdl (bool): Whether to enable VisualDL during training. Default: True. save_dir (str|None): Directory to store model snapshots and logs. If None, use a default setting. Default: None. Returns: paddlex.repo_apis.base.utils.subprocess.CompletedProcess NotImplementedError) r' batch_size learning_rate epochs_itersipsdeviceZ resume_pathdy2stamp num_workersuse_vdlsave_dirkwargsrrrres1zBaseModel.traincKr,)a Evaluate a model. Args: weight_path (str): Path of the weights to initialize the model. batch_size (int|None): Number of samples in each mini-batch. If multiple devices are used, this is the batch size on each device. If None, use a default setting. Default: None. ips (str|None): If not None, enable multi-machine evaluation mode. `ips` specifies Paddle cluster node ips, e.g., '192.168.0.16,192.168.0.17'. Default: None. device (str): A string that describes the device(s) to use, e.g., 'cpu', 'gpu', 'gpu:1,2'. Default: 'gpu'. amp (str): Optimization level to use in AMP training. Choices are ['O1', 'O2', 'OFF']. Default: 'OFF'. num_workers (int|None): Number of subprocesses to use for data loading. If None, use a default setting. Default: None. Returns: paddlex.repo_apis.base.utils.subprocess.CompletedProcess r-)r' weight_pathr/r2r3r5r6r9rrrrs zBaseModel.evaluatecKr,)a2 Make prediction with a pre-trained model. Args: weight_path (str): Path of the weights to initialize the model. input_path (str): Path of the input file, e.g. an image. device (str): A string that describes the device to use, e.g., 'cpu', 'gpu'. Default: 'gpu'. save_dir (str|None): Directory to store prediction results. If None, use a default setting. Default: None. Returns: paddlex.repo_apis.base.utils.subprocess.CompletedProcess r-)r'r: input_pathr3r8r9rrrrzBaseModel.predictcKr,)a" Export a pre-trained model. Args: weight_path (str): Path of the weights to initialize the model. save_dir (str): Directory to store the exported model. Returns: paddlex.repo_apis.base.utils.subprocess.CompletedProcess r-)r'r:r8r9rrrrs zBaseModel.exportcKr,)a2 Make inference with an exported inference model. Args: model_dir (str): Path of the exported inference model. input_path (str): Path of the input file, e.g. an image. device (str): A string that describes the device(s) to use, e.g., 'cpu', 'gpu'. Default: 'gpu'. save_dir (str|None): Directory to store inference results. If None, use a default setting. Default: None. Returns: paddlex.repo_apis.base.utils.subprocess.CompletedProcess r-)r'Z model_dirr;r3r8r9rrrrr<zBaseModel.inferc Kr,)ak Perform quantization aware training (QAT) and export the quantized model. Args: weight_path (str): Path of the weights to initialize the model. batch_size (int|None): Number of samples in each mini-batch. If multiple devices are used, this is the batch size on each device. If None, use a default setting. Default: None. learning_rate (float|None): Learning rate of QAT. If None, use a default setting. Default: None. epochs_iters (int|None): Total epochs of iterations of model training. If None, use a default setting. Default: None. device (str): A string that describes the device(s) to use, e.g., 'cpu', 'gpu'. Default: 'gpu'. use_vdl (bool): Whether to enable VisualDL during training. Default: True. save_dir (str|None): Directory to store the results. If None, use a default setting. Default: None. Returns: tuple[paddlex.repo_apis.base.utils.subprocess.CompletedProcess] r-) r'r:r/r0r1r3r7r8r9rrrr s#zBaseModel.compressionc c|j}|jd}d|j|g}|d}tjsUtjt d*}t j ||}t |dddWdn1s;wY|VWddS1sNwYdSt j t |}t |dddWdn1sowY|VdS)Nr_z.ymldirwutf-8encoding r)rjoin__name__lowerrDEBUGtempfileTemporaryDirectoryrospathopen)r'clsrtagZyaml_file_nametdrMrrr_create_new_config_file$ " z!BaseModel._create_new_config_filec cr=)Nrr>z _test.jsonr?rArBrCrE)r'rOrrPZjson_file_namerQrMrrr_create_new_val_json_file#rSz#BaseModel._create_new_val_json_filecC|jddS)zsupported apissupported_apisNrgetr'rrrrV5zBaseModel.supported_apiscC|j|jjdddS)zsupported train optsrapi_nameNrrX_API_SUPPORTED_OPTS_KEY_PATTERNformatrYrrrsupported_train_opts:zBaseModel.supported_train_optscCr[)zsupported evaluate optsrr\Nr^rYrrrsupported_evaluate_optsArbz!BaseModel.supported_evaluate_optscCr[)zsupported predcit optsrr\Nr^rYrrrsupported_predict_optsHrbz BaseModel.supported_predict_optscCr[)zsupported infer optsrr\Nr^rYrrrsupported_infer_optsOrbzBaseModel.supported_infer_optscCr[)zsupported copression optsr r\Nr^rYrrrsupported_compression_optsVrbz$BaseModel.supported_compression_optscCrU)zsupported dataset typessupported_dataset_typesNrWrYrrrrg]rZz!BaseModel.supported_dataset_typescCs&t|dkrtd|ddSdS)Nrz'Unconsumed keyword arguments detected: .)lenrr%)r9rrr_assert_empty_kwargsbs zBaseModel._assert_empty_kwargscsfdd}fdd}j}|durtj}ntj}jD]#}t|}||vr7||}t||q!tjrD||}t||q!dS)Ncstfdd}|S)Ncs j}j}t|d|d)Nz does not support `z`.)r#rGr)argsr9rr]) bnd_methodr'rr_unavailable_apims zJBaseModel._patch_apis.._make_unavailable.._unavailable_api) functoolswraps)rlrmrY)rlr_make_unavailablelsz0BaseModel._patch_apis.._make_unavailablecstfdd}j}g|dkrGj}|durEd|vr+t|dddd|vr8t|dd|vrEt|d|S|d krpj}|durnd|vrat|dddd|vrnt|d|S|d krj }|durd|vrt|dd d|S|d krj }|durd|vrt|dd d|S|d krĈj }|durd|vrˆt|ddd|SS)Ncsvtj}|j|i|}|j}|jD]}|j|vr)|j|j ur)|j||j<qD]}| |q,|i|SN) inspect Signature from_callablebind arguments parametersvaluesr#defaultemptycheck)rkr9sigZbnd_argsZ args_dictpr{rlZchecksrr_api_with_prechecksxs   zJBaseModel._patch_apis.._add_prechecks.._api_with_prechecksrr3T)check_mcr4r5rrFrr ) rnrorGraappend _CheckDevice _CheckDy2St _CheckAMPrcrdrerf)rlrr]optsrYr~r_add_precheckswsR z-BaseModel._patch_apis.._add_prechecks)rVset_API_FULL_LISTgetattrsetattrrZ CHECK_OPTS)r'rprrVZ avail_api_setr]apiZ decorated_apirrYrr&ks  9     zBaseModel._patch_apisrq) NNNNr*NFr+NTN)NNr*r+N)r*N)NNNr*TN)rG __module__ __qualname____doc__rr_r"abcabstractmethodrrrrrr contextlibcontextmanagerrRrTcached_propertyrVrarcrdrerfrg staticmethodrjr& __classcell__rrr(rr>sx 2 !    $          ) metaclassc@s$eZdZdZdZddZddZdS) _CheckFailedTcCs||_||_||_dSrq)arg_namearg_val legal_vals)r'rrrrrrr"s z_CheckFailed.__init__cCsd|jd|jd|jS)N`z*` is expected to be one of or conforms to z , but got )rrrrYrrr__str__sz_CheckFailed.__str__N)rGrrrZcheck_failed_errorr"rrrrrrs  rcs(eZdZdZfddZddZZS)_APICallArgsCheckercst||_dSrq)r!r"r)r'rr(rrr"s  z_APICallArgsChecker.__init__cCr,)r{r-)r'rkrrrr{sz_APICallArgsChecker.checkrGrrrr"r{rrrr(rrs rcs*eZdZdZdfdd ZddZZS)rFcst|||_dSrq)r!r"r)r'rrr(rrr"s  z_CheckDevice.__init__cCs4d|vsJ|d}|durt|\}}|js%||jvr#td||jdS|dkr7||jvr5td||jdS|d}|d}|d}t|dkrh||jvrb||jvrd||jvrftd||jdSdSdSd|vsnJ|ddur||jvrtd||jdS||jvr||jvrtd||jdSdS dS) r{r3Nr*Z_n1c1Z_n1cxZ_nxcxr r2)rrrrri)r'rkr3Z device_typeZdev_idsZ n1c1_descZ n1cx_descZ nxcx_descrrrr{sD                z_CheckDevice.check)Frrrr(rrsrc@eZdZdZddZdS)rcCsxd|vsJ|d}t|jtr!t|jdksJt|jd}nt|j}|dur9|r5|s7td||gdSdS dS)r{r4r rN) isinstancerlistriboolr)r'rkr4Z support_dy2strrrr{s   z_CheckDy2St.checkNrGrrrr{rrrrr rc@r)rcCsJd|vsJ|d}|dur"|dkr||jvr td||jdSdS dS)r{r5Nr+)rr)r'rkr5rrrr{+s z_CheckAMP.checkNrrrrrr(rr)NN)%rrrnrrrLrJutilsrrZ utils.cacherZ utils.devicerZ utils.errorsrrZ utils.miscr rrr registerr r r__all__rrABCMetar Exceptionrobjectrrrrrrrrs2       4