j3ddlmZddlmZddlmZddlZddlZddl m Z m Z m Z ddl mZmZmZddlmZmZmZmZddlmZdd lmZdd lmZd,dZd-dZd.d/dZ d0d1d"Zd2d3d$Z d2d4d'Z!d5d6d*Z"d+Z#dS)7) annotations)partial)PathN)DetectPosePose26)LINUXLOGGERMACOS)check_apt_requirementscheck_requirements check_versionis_sudo_available)attempt_download_asset)spaces_in_path) make_anchorsmodeltorch.nn.Modulereturnc X|D]}t|tsddl}|t ||_t|trC|ttt|tu||_ |S)z\A wrapper for TensorFlow export compatibility (TF-specific handling is now in head modules).rN) is_pose26) modules isinstancertypes MethodType_tf_decode_boxes_get_decode_boxesrr_tf_kpts_decodetyper kpts_decode)rmrs f/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/ultralytics/utils/export/tensorflow.py tf_wrapperr#s ]]__gg!V$$   #../?CC a   g!,,W_PTUVPWPW[aPa-b-b-bdeffAM Lxdict[str, torch.Tensor] torch.TensorcB|ddj}|d}|jdkrL|js |j|kr:dt|d|jdD\|_|_||_|dd\}}tj||||g|j  d dd }|j|jd|zz }| | ||z|j d|d d d dfz}|S) z,Decode bounding boxes for TensorFlow export.featsrboxesimxc3BK|]}|ddVdS)rN) transpose).0as r" z#_tf_decode_boxes..%s0%l%lAakk!Q&7&7%l%l%l%l%l%lr$?devicer-N)shapeformatdynamicrstrideanchorsstridestorchtensorr6reshape decode_bboxesdfl unsqueeze) selfr%r7r*grid_hgrid_w grid_sizenormdboxs r"rr s' gJqM E gJE {eu1D1D%l%laPWjZ^ZegjAkAk%l%l%l" dl 1Q3ZNFF ffff=elSSS[[\]_`bcddI <4;q>I5 6D   dhhuoo4dl6L6LQ6O6ORVWXWXWXZ\[\Z\W\R]6] ^ ^D Kr$FkptsrboolcF|jd}|jd}|j|g|jdR}|jdd\}}tj||g|jddd}|j|jd|zz } |r|ddddddf|j zn|ddddddfdz|j d z z| z} |d kr8tj | |dddddd f fd} | ||j dS) z'Decode keypoints for TensorFlow export.r-rr3r4r5Ng@r2) kpt_shaper7viewr=r>r6r?r<r:r;catsigmoidnk) rCrIrndimbsyrDrErFrGr0s r"rr.sC >! D AB "*t~*r***AZ!_NFF ff-ah???GG1aPPI <4;q>I5 6D)2 b!AAAqqq"1"H+ $ $111aaa!8s9Jdl]`N`9afjjA qyy Iq!AAAqqq!A#I,..0011 5 5 66"dgr " ""r$ onnx_filestr output_dir Path | strint8imagesnp.ndarray | Nonedisable_group_convolutionprefixc tj} ddl}n##t$rt dddl}YnwxYwt ddddt rdnd zd d d |rd nddf dtjd|d|j dt|j ddddt|}td}| st|dddd} |rQ|dz } |J|ddtjt#| |d| gd ggggd!ggggg} ddl} t'| jd"sddlfd#} | | j_ddl} tj|d$| j d| |t#|dd%|| | d|& }|r| d'|d(D]K}|||jd)d*|jzL|d+D]}||S),adConvert an ONNX model to TensorFlow SavedModel format using onnx2tf. Args: onnx_file (str): ONNX file path. output_dir (Path | str): Output directory path for the SavedModel. int8 (bool, optional): Enable INT8 quantization. Defaults to False. images (np.ndarray | None, optional): Calibration images for INT8 quantization in BHWC format. disable_group_convolution (bool, optional): Disable group convolution optimization. Defaults to False. prefix (str, optional): Logging prefix. Defaults to "". Returns: (keras.Model): Converted Keras model. Notes: - Auto-installs tensorflow, onnx2tf, and all required dependencies if not present. - Downloads calibration data if INT8 quantization is enabled. - Removes temporary files and renames quantized models after conversion. rNztensorflow>=2.0.0,<=2.19.0ztf_keras<=2.19.0zsng4onnx>=1.0.1zonnx_graphsurgeon>=0.3.26zai-edge-litert>=1.2.0z,<1.4.0rVzonnx>=1.12.0,<2.0.0zonnx2tf>=1.26.3,<1.29.0zonnxslim>=0.1.71zonnxruntime-gpu onnxruntimez protobuf>=5z---extra-index-url https://pypi.ngc.nvidia.com)cmds ! starting export with tensorflow ...z>=2.0.0 tensorflowTz6https://github.com/ultralytics/ultralytics/issues/5161)nameverbosemsgz6calibration_image_sample_data_20x128x128x3_float32.npyz.zip)unzipdeletez&tmp_tflite_int8_calibration_images.npy)parentsexist_okr\)rrr)rnrnfloat32_to_bfloat16cndd|d}|dz S)zBConvert float32 to bfloat16 (truncates lower 16 bits of mantissa).z=Iz=fr)unpackpack)fvalivalstructs r"roz-onnx2saved_model..float32_to_bfloat16s3==v{{4'>'>??BD2: r$z% starting TFLite export with onnx2tf error) input_onnx_file_pathoutput_folder_pathnot_use_onnxsim verbosityoutput_integer_quantized_tflite!custom_input_op_name_np_data_pathenable_batchmatmul_unfoldoutput_signaturedefsr^) missing_okz*_dynamic_range_quant.tflite_dynamic_range_quant_int8z%*_integer_quant_with_int16_act.tflite) r=cuda is_availablerf ImportErrorr r r info __version__rrexistsrmkdirnpsaverX onnx.helperhasattrhelperrvroonnx2tfconvertunlinkrglobrename with_namestemreplacesuffix)rWrYr[r\r^r_rtf onnx2tf_filenp_datatmp_fileonnxror keras_modelfilervs @r"onnx2saved_modelr=s?4 : " " $ $D     7888    ' #E'Ayyr B ! % !% 8  =  =     KQVQQbnQQQRRR   D j!!JPQQL    O,444DNNNNG U HH     TD  9 9 9 GCMM6 * * * (yyyk]O?P>Q=RSTG 4; 5 6 6>       +> 'NNN K6XX@SXXXYYY//&z??(,*1&*(!";"  K 4((($$%CDD j jD KKty'8'89OQX'Y'Y\`\g'ghh i i i i$$%LMM  D KKMMMM s&AA output_filecddl}ddlm}tjd|d|jd|fd}||j dj j dj }||}|j t|}|j|j t#|j|jd t#|S) aConvert a Keras model to TensorFlow GraphDef (.pb) format. Args: keras_model (keras.Model): Keras model to convert to frozen graph format. output_file (Path | str): Output file path (suffix will be changed to .pb). prefix (str, optional): Logging prefix. Defaults to "". Returns: (str): Path to the exported ``.pb`` file. Notes: Creates a frozen graph by converting variables to constants for inference optimization. rN)!convert_variables_to_constants_v2rcrdrec|S)N)r%rs r"zkeras2pb..skk!nnr$F)graph_or_graph_deflogdirrgas_text)rf0tensorflow.python.framework.convert_to_constantsrr rrfunctionget_concrete_function TensorSpecinputsr7dtypegraph as_graph_defrio write_graphrXparentrg)rrr_rrr! frozen_funcs` r"keras2pbrsbbbbbb KQVQQbnQQQRRR ,,,,--A  k.@.C.I;K]^_K`Kf g ghhA33A66K""$$${##KE&,S9K5L5LS^Scmr {  r$ tflite_file str | Pathcddl}d}d}ts Jd||||j|jdjdkrgt jd|d |trd nd }|d d |dd|dfD]}||ddtdg||dddj dd}t jd|d|dd|d|d} t j|d| d|| dtt|t|jdz S) aPConvert a TensorFlow Lite model to Edge TPU format using the Edge TPU compiler. Args: tflite_file (str | Path): Path to the input TensorFlow Lite (.tflite) model file. output_dir (str | Path): Output directory path for the compiled Edge TPU model. prefix (str, optional): Logging prefix. Defaults to "". Returns: (str): Path to the exported Edge TPU model file. Notes: Auto-installs the Edge TPU compiler if not found. The function compiles the TFLite model for optimal performance on Google's Edge TPU hardware accelerator. rNzedgetpu_compiler --versionz'https://coral.ai/docs/edgetpu/compiler/z$export only supported on Linux. See T)stdoutstderrshellrcz< export requires Edge TPU compiler. Attempting install from zsudo rVzmkdir -p /etc/apt/keyringszCcurl -fsSL https://packages.cloud.google.com/apt/doc/apt-key.gpg | z6gpg --no-tty --dearmor -o /etc/apt/keyrings/google.gpgzvecho "deb [signed-by=/etc/apt/keyrings/google.gpg] https://packages.cloud.google.com/apt coral-edgetpu-stable main" | z.tee /etc/apt/sources.list.d/coral-edgetpu.list)rcheckzedgetpu-compiler)rcapture_outputrr-)maxsplitrLz( starting export with Edge TPU compiler rezedgetpu_compiler --out_dir "zS" --show_operations --search_delegate --delegate_search_step 30 --timeout_sec 180 "" running ''rz_edgetpu.tflite) subprocessr runDEVNULL returncoder rrr rdecodersplitrXrr) rrYr_r check_cmdhelp_urlsudocvercmds r"tflite2edgetpurs-I8H CCCCCCC5~~i (::CU]a~bbmqrrr ggg]egghhh+--5ww2 / / / ORV O O O zFJ z z z  6 6A NN1DN 5 5 5 5 23444 ..$t4. P P W ^ ^ ` ` g gqr g s stv wC KMVMMSMMMNNN          K6++S+++,,,NN3dN### tJT+%6%6%;"L"L"LL M MMr$pb_filehalfc dddl}tdddl}ddl}t jd|d|jd|}t|d5} | | dddn #1swxYwYd t|} t jd|d | |rd n|rd nd } t|5} t|5} d | d| d| d| d }t j|d|d||ddddn #1swxYwYdddn #1swxYwYd|vrt j|d|dt#|S)aConvert a TensorFlow GraphDef (.pb) model to TensorFlow.js format. Args: pb_file (str): Path to the input TensorFlow GraphDef (.pb) model file. output_dir (str): Output directory path for the converted TensorFlow.js model. half (bool, optional): Enable FP16 quantization. Defaults to False. int8 (bool, optional): Enable INT8 quantization. Defaults to False. prefix (str, optional): Logging prefix. Defaults to "". Returns: (str): Path to the exported TensorFlow.js model directory. Notes: Auto-installs tensorflowjs if not present. Uses tensorflowjs_converter command-line tool for conversion. Handles spaces in file paths and warns if output directory contains spaces. rN tensorflowjsrcz# starting export with tensorflowjs rerb,z output node names: z--quantize_float16z--quantize_uint8rVz6tensorflowjs_converter --input_format=tf_frozen_model z --output_node_names=z "z" "rrrTr z8 your model may not work correctly with spaces in path 'z'.)rr rfrr rrGraphropenParseFromStringreadjoin gd_outputsrrwarningrX)rrYrr[r_rrtfjsgdfoutputs quantizationfpb_f_rs r"pb2tfjsrs"~&&& KUVUU@PUUUVVV  " "B gt  % 16688$$$%%%%%%%%%%%%%%%hhz"~~&&G K:V::::;;;+/W''45W5G5GUWL  (D.*D*D( k.: k kQX k k\` k keg k k k   v/////000s$''' (((((((((((((((((((((((((((((( j&hhZdhhhiii z??sI2(B&&B*-B*E:AE# E:#E' 'E:*E' +E::E>E>c  gg}}|jD]6}||j||j7t dt t|t|z DS)z3Return TensorFlow GraphDef model output node names.c3JK|]}|d|dVdS)NoOpz:0N) startswith)r/r%s r"r1zgd_outputs..*s=ggqRSR^R^_eRfRfgQ(((ggggggr$)nodeappendrgextendinputsortedlistset)r name_list input_listrs r"rr$szI&&###$*%%%% ggDY#j//)I$J$Jggg g ggr$)rrrr)r%r&rr')F)rIr'rrJrr')FNFrV) rWrXrYrZr[rJr\r]r^rJr_rX)rV)rrZr_rXrrX)rrrYrr_rXrrX)FFrV) rrXrYrXrrJr[rJr_rXrrX)$ __future__r functoolsrpathlibrnumpyrr=ultralytics.nn.modulesrrrultralytics.utilsr r r ultralytics.utils.checksr r rrultralytics.utils.downloadsrultralytics.utils.filesrultralytics.utils.talrr#rrrrrrrrr$r"rs#""""" 77777777772222222222qqqqqqqqqqqq>>>>>>222222......         # # # # #$ $&+ iiiiiX:.N.N.N.N.Nb*****Zhhhhhr$