]jgJ ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZddlmZddlmZddlZddlZddlmZddlmZddlmZdd lmZdd lmZej@e!Z"d d hZ#d hZ$edejJjMejNZ(ejRddfejRddfejRddfejRddfejRddfejRddfejRddfge_*GddejVZ,dZ-d Z.d0d!Z/ d1d"Z0d#Z1d$Z2d%Z3d&Z4d'Z5d(Z6d)Z7d2d*Z8d+Z9d,Z:d-Z;d3d.Z z([^\.])(\.)([\]\)}>"\']*)\s*$z\1 \2\3 z[?!]z([^'])' z\1 ' ceZdZddZy)ExpandFullPathNc t||jtjj tjj |yN)setattrdestospathabspath expanduser)selfparser namespacevalues option_strings J/root/env/lib/python3.12/site-packages/label_studio_sdk/converter/utils.py__call__zExpandFullPath.__call__3s. 499bggoobgg6H6H6P&QRr)__name__ __module__ __qualname__r!r"r rr2sSr"rcd}g}|jD]7}t|r%|j||f|t|dzz }3|dz }9|S)Nr)splitlenappend)text tok_startouttoks r tokenizer07s[I Czz| s8 JJY' ( SA %I NI  Jr"ctj|Dcgc]}|}}|Dcgc] \}}||||f}}}|rGt|Dcgc]*}|jdduxr|jddu,c}rt t |t d}|jd}|d}|ddz } d} gg} } |D]\}} | j|| t|zdz }| }|| kDr+|r)|jd}|d}|ddz } d} | | krn|r)|r||ks|jds| jd||kr"| |k\r| j| |ddzd } | jd| | fS|Dcgc]\}}| } }}dgt| z} | | fScc}wcc}}wcc}wcc}}w) Nstartend)keyrr(zB-labelsOzI-) r span_tokenizeallgetlistsortedrpopr+r*)r,spanstokentoken_index_tuplesr2r3tokens_and_idxspan span_startspan_endprefixtokenstags token_start token_endtoken_start_ind_s r create_tokens_and_tagsrKCs12@@FBTT:5#tE#.TNT   HHW T ) IdhhuoT.I I  VEz'':;<yy|'] ;?2"0 ! E; MM% c%j(1, *O ) 99Q.>x.H  YYx3B7==eD(v||++H56"#>( #++JJL3x'8'8'<'<'>'H'H'J#K#R#R#TT)+bq""   ww||J1H 77>>( # 299cH9UV  S!A   - & 199% &ww||BGG,,Z8"'':J:J8:TUU O  & &s 7O99Pc@tj|jSr)rrxsize) image_paths r get_image_sizers ::j ! & &&r"ctj|}|j\}}t|j }|||fSr)rrxrr*getbands)riwhcs r get_image_size_and_channelsrs8 :A 66DAq AJJLA a7Nr"ctj|d5}|jt|j z cdddS#1swYyxYw)Nrr^)waverx getnframesfloat getframerate) audio_pathfs r get_audio_durationrsA :C (8A||~ann&6 77888s *A  Acntjj|stj|yyr)rrrQmakedirs)rs r ensure_dirrs# 77>>( # H $r"c |siSd}d}d} tj|}iit t }}}|jD]} || r| j| jdjdd} i} | jjdd k(r| jjd rd | jd d } n[| jjd rd| jd d } n-| jjdrd| jdd } | r| | d<| || jd<nC|| r;| j| jdjdd|| jd<| jtvrQ|| |} | ^| jjdxs| jjd} | sMtjdj!tj"| dj%ddt | j|| | <|j'D]e\}} g| d<| dD]:}||vrtjd|d|d $| dj)||<t+||| d!<||| d"<g|S#tj$r}tt |d}~wwxYw)#a :param config_string: Label config string :return: structured config of the form: { ".name": { "type": "ControlTag", "to_name": [".name", ".name"], "inputs: [ {"type": "ObjectTag1", "value": ".value"}, {"type": "ObjectTag2", "value": ".value"} ], "labels": ["Label1", "Label2", "Label3"] // taken from "alias" if exists or "value" } cr|jjdxr|jjdS)Nnamevalue)attribr9tags r _is_input_tagz#parse_config.._is_input_tags'zz~~f%A#**..*AAr"c|jjdxr/|jjdxr|jtvS)NrtoName)rr9r_NOT_CONTROL_TAGSrs r _is_output_tagz$parse_config.._is_output_tags> JJNN6 " 1 x( 100 r"cr|} |j}|y|jjd}||vr|S5)Nr) getparentrr9)routputsparentrs r _get_parent_output_tag_namez1parse_config.._get_parent_output_tag_namesE%%'F~==$$V,Dw r"Nr,)typeto_name perRegiontrue whenTagNamer)rrwhenLabelValuelabelwhenChoiceValuechoice conditionalsrr$)rraliaszDInspecting tag {tag_name}... found no "value" or "alias" attributes.unicodeencoding2)tag_nameinputsrzto_name=z" is specified for output tag name=z', but we can't find it among input tagsr5 labels_attrs)r fromstringXMLSyntaxError ValueErrorrkrdictiterrrr)r9lstrip _LABEL_TAGSrerfrgtostringstripitemsr+r:) config_stringrrrxml_treeerrr5rtag_infor parent_name actual_value output_taginput_tag_names r parse_configrs  B  !##M2!"k$&7VGF}}&E #  #CJJx4H4N4Ns4STHLzz~~k*f4::>>-0,13::m;T#ULZZ^^$45 ' # +; <$LZZ^^$56 ( # +< =$L+7(*2GCJJv& ' 3 G,33C8*F3::f% & 77+ % 1#w?  "::>>'2McjjnnW6ML Zaa!&i!H!N!N!PQTRT!Ub 594D{#L1M&EN!(  6 H&y1 >NV+ ~..PQ[P\]<< X  % %f^&< = >"&"45#)*#5  6 No   !Q  !sKK0K++K0c t|t|k(sJtdtjtj|tj |dtj|tj |dz zS)z.https://en.wikipedia.org/wiki/Shoelace_formulag?r()r*rnpabsdotroll)xys r get_polygon_arearCsb q6SV   rvvbffQ1 62771a=9QQRR SSr"ct|t|k(sJt|t|t|t|f\}}}}||||z ||z gSr)r*minmax)rrx1y1x2y2s r get_polygon_bounding_boxrKsU q6SV  VSVSVSV3NBB BGR"W %%r"c|d}t|tr|jd|}|St|tr|r|St |S)z)Get annotator id or email from annotation completed_byemail) isinstancerr9intrk)itemrint_id annotators r get_annotatorrRsH^$I)T"MM'73 )S!f y>r"cd}t|dd5}|dk7rJ|jd}|jr'|dk(r dddy|d k(r dddy dddy dddy #1swYy xYw) Nrrzutf-8rrWr({r[r:invalidempty)rxreadisspace)r{charrs r get_json_root_typer_s D hg .!bj66!9D||~s{s{$%* +* s.A+A+A++A4cg}d}|D]}t|}|jd}|dk(r)t|ddk(r|j|ddM|dk(r)t|ddk(r|j|dd{|d vr"|j|j d ||j||d vrt|dk(r|dS|S) z :param v: list of regions or results :return: label name as is if there is only 1 item in result `v`, else list of label names NrChoiceschoicesr(rTextArear,)Chat chatmessager)rr)rr<r*r+r9)vr.tag_typerjs r prettify_resultrys CH   QK55= y S9%6!%; JJq|A '  #AfI!(; JJqy| $ 0 0 JJquu]A. / JJqM !88SX]3q6SPSSr"cd|vr d|vrd|vrd|vsy |d}|d}|d}|d}||||y||dz zdz }||dz zdz }|dz }|dz }||||fS#ttf$rYywxYw)a Convert LS annotation to Yolo format. Args: label (dict): Dictionary containing annotation information including: - width (float): Width of the object. - height (float): Height of the object. - x (float): X-coordinate of the top-left corner of the object. - y (float): Y-coordinate of the top-left corner of the object. Returns: tuple or None: If the conversion is successful, returns a tuple (x, y, w, h) representing the coordinates and dimensions of the object in Yolo format, where (x, y) are the center coordinates of the object, and (w, h) are the width and height of the object respectively. rrwidthheightNd) TypeErrorr)rrrrrs r convert_annotation_to_yolors" 5LSE\g.>8uCT  #J #J 'N (O 9 QY!) QY#  QY#  G G aA: z "sA AA('A(cd|vrd|vrd|vrd|vr d|vrd|vrd|vsy|d|d}}|dd z |z}|dd z |z}|dd z |z}|dd z |z}tj|jdd }tj|tj|} } ||f||| zz||| zzf||| zz|| zz ||| zz|| zzf||| zz ||| zzfg} |r| D cgc]} | d |z | d |z fc} S| Scc} w) av Convert LS annotation to Yolo OBB format. Args: label (dict): Dictionary containing annotation information including: - original_width (int): Original width of the image. - original_height (int): Original height of the image. - x (float): X-coordinate of the top-left corner of the object in percentage of the original width. - y (float): Y-coordinate of the top-left corner of the object in percentage of the original height. - width (float): Width of the object in percentage of the original width. - height (float): Height of the object in percentage of the original height. - rotation (float, optional): Rotation angle of the object in degrees (default is 0). normalize (bool, optional): Whether to normalize the coordinates to the range [0, 1] (default is True). Returns: list of tuple or None: List of tuples containing the coordinates of the object in Yolo OBB format. Each tuple represents a corner of the bounding box in the order: (top-left, top-right, bottom-right, bottom-left). original_widthoriginal_heightrrrrrotationNrrr()mathradiansr9cossin) r normalize org_width org_heightrrrrr r r coordscoords r convert_annotation_to_yolo_obbrst, E!  & 5L 5L u    % !"23U;L5MzI c S9$A c S:%A gy(A h# *A||EIIj!45Hxx!488H#5C A QWa!c'k" QWq3w AG a#g 56 QWa!c'k" FKQR%qI%uQx*'<=RR Ss'Dctj|tjjd}tj|dddf}tj|dddf}tj j |d|dz }tj j |d|dz }|d|dz }|d |d z } tjtj| |} ||d z tjtj| zz |d z tjtj| zz} ||d z tjtj| zz |d z tjtj| zz } | |z d z} | |z d z}||z d z}||z d z}| |||| ||d S)ai Convert YOLO Oriented Bounding Box (OBB) format to Label Studio format. Args: xyxyxyxy (list): List of 8 float values representing the absolute pixel coordinates of the OBB in the format [x1, y1, x2, y2, x3, y3, x4, y4]. original_width (int): Original width of the image. original_height (int): Original height of the image. Returns: dict: Dictionary containing the converted bounding box with the following keys: - x: X-coordinate of the top-left corner of the bounding box in percentage. - y: Y-coordinate of the top-left corner of the bounding box in percentage. - width: Width of the bounding box in percentage. - height: Height of the bounding box in percentage. - rotation: Rotation angle of the bounding box in degrees. )dtype)r[rNrr()r(r)rr)r(r()rr(rr)rrrrr rr) rarrayfloat64reshapemeanlinalgnormdegreesarctan2r r r )xyxyxyxyrrrcenter_xcenter_yrrdxdyr top_left_x top_left_yrrs r convert_yolo_obb_to_annotationr&s&XXhbjj 1 9 9& AFwwvad|$Hwwvad|$H IINN6!9vay0 1E YY^^F1Iq 1 2F t $B t $B 2::b"%&A  19rzz!}- - . A: 1 . . /  19rzz!}- - . A: 1 . . / n $+A o %,A ^ #s *E&# -F (* r")NN)NNFNT)NF)T)>argparserprmrwloggingr rrerhrar collectionsrriroperatorr urllib.parsernumpyrruPILrlxmlrnltk.tokenize.treebankr Alabel_studio_sdk._extensions.label_studio_tools.core.utils.paramsr =label_studio_sdk._extensions.label_studio_tools.core.utils.ior getLoggerr#rerrrrseprcompile PUNCTUATIONActionrr0rKrUrrrrrrrrrrrrrr&r&r"r r8s  #!8UY   8 $!  $)@ RZZ !9-RZZ W%RZZ H%RZZ z* 34RZZ*%RZZ h' %!SX__S 3l &?D'8  `FT& 4T*"J4n=r"