o "j@sddlZddlZddlmZddlZddlZddlmZddl m Z m Z m Z m Z mZmZddlmZeeejddZe ZeZe jZe Ze Ze Zdd Zd d Zd efd edededededede deefddZ!ee_"ee_#dS)N)Set)core)AnalysisConfig PaddleDTypePaddleInferPredictorPaddleInferTensor PaddlePlaceconvert_to_mixed_precision_bind) get_loggerz&%(asctime)s-%(levelname)s: %(message)s)fmtcCsFt|tjst|trt|dkrt|dtr||dStd)zA Support input type check based on tensor.copy_from_cpu. rzHIn copy_from_cpu, we only support numpy ndarray and list[str] data type.N) isinstancenpZndarraylistlenstrZ_copy_from_cpu_bind TypeErrorselfdatarY/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/inference/wrapper.pytensor_copy_from_cpu-s rcCsTt|tjr ||dSt|tjr||dSt|tjjj r&t dt d)zG Support input type check based on tensor.share_external_data. zThe interface 'share_external_data' can only be used in dynamic graph mode. Maybe you called 'paddle.enable_static()' and you are in static graph mode now. Please use 'copy_from_cpu' instead.z=In share_external_data, we only support Tensor and LoDTensor.N) r rZ LoDTensorZ_share_external_data_bindpaddleTensorZ'_share_external_data_paddle_tensor_bindbaseZ frameworkVariablerrrrrtensor_share_external_data;s  rT model_file params_filemixed_model_filemixed_params_filemixed_precisionbackend keep_io_types black_listc Ks|tjurtstdtj|} t |dkr tj|n| } tj | s-t | | dt } t||||||||| dS)a Convert a fp32 model to mixed precision model. Args: model_file: fp32 model file, e.g. inference.pdmodel. params_file: fp32 params file, e.g. inference.pdiparams. mixed_model_file: The storage path of the converted mixed-precision model. mixed_params_file: The storage path of the converted mixed-precision params. mixed_precision: The precision, e.g. PrecisionType.Half. backend: The backend, e.g. PlaceType.GPU. keep_io_types: Whether the model input and output dtype remains unchanged. black_list: Operators that do not convert precision. kwargs: Supported keys including 'white_list'. - white_list: Operators that do convert precision. zOYou shoule use PaddlePaddle compiled with GPU when backend set to PlaceType.GPUr white_listN) PlaceTypeZGPUrZis_compiled_with_cuda_loggererrorospathdirnamerexistsmakedirsgetsetr ) rrrr r!r"r#r$kwargsZmixed_model_dirnameZmixed_params_dirnamer%rrrconvert_to_mixed_precisionOs.     r1)$loggingr)typingrnumpyr rZ paddle.baserZpaddle.base.corerrrrrr Zpaddle.base.log_helperr __name__INFOr'ZDataTyper&Z PrecisionZ PrecisionTypeZConfigrZ Predictorrrr/rboolr1Z copy_from_cpuZshare_external_datarrrrsP     8