^j/8ddlmZddlZddlmZddlmZmZddlZddl Z ddl m Z ddlmZmZddlmZddlmZddlmZdd lmZmZdd lmZ d dd Z d dd Z d dd Z d ddZddZ ddZ!ddZ"y)) annotationsN)Path)Element SubElement)parsetostring) parseString)approximate_mask_with_polygons) Detections)polygon_to_maskpolygon_to_xyxy)list_files_with_extensionscztd}t|d}||_|dz}t|d}t|d}tt |d|_t|d}tt |d|_t|d}tt |d |_t|d } tt |d | _|i|dz}t|d } t |d D]H\} } | \} }t| d| }t| |_t| d| }t||_J|S)aBuild a Pascal VOC ```` XML element for one detection. Coordinates are converted to 1-indexed Pascal VOC convention before writing. The input arrays are never mutated; new arrays are allocated for the offset. Args: xyxy: Bounding box in zero-indexed pixel coordinates ``[x1, y1, x2, y2]``. Shape ``(4,)``. name: Class label string written to the ```` child element. polygon: Optional segmentation polygon in zero-indexed pixel coordinates. Shape ``(N, 2)``. Returns: An XML ``Element`` rooted at ```` containing ````, ````, and optionally ```` children. Examples: >>> import numpy as np >>> from supervision.dataset.formats.pascal_voc import object_to_pascal_voc >>> elem = object_to_pascal_voc(np.array([0, 0, 9, 9]), name="cat") >>> elem.find("bndbox/xmin").text '1' >>> elem.find("bndbox/xmax").text '10' objectnamebndboxxminryminxmaxymaxpolygon)startxy)rrtextstrint enumerate)xyxyrrroot object_namerrrrrobject_polygonindexpoint x_coordinate y_coordinaterrs q/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/supervision/dataset/formats/pascal_voc.pyobject_to_pascal_vocr+s;< 8 DT6*KK !8D h 'F ff %DCQL!DI ff %DCQL!DI ff %DCQL!DI ff %DCQL!DIA+#D)4%gQ7 'LE5). &L,>Qug;7A&AF>Qug;7A&AF  ' Kcf|\}}} td} t| d} d| _t| d} || _t| d} t| d}d|_t| d}t|d }t||_t|d }t||_t|d }t| |_t| d }d |_|D]\}}}}}}| t dt |t tjfst dt|d||}|Bt||||}|D]-}t|}t|||}| j|/t||}| j|ttt| j!d}|S)a Converts Detections object to Pascal VOC XML format. Args: detections: A Detections object containing bounding boxes, class ids, and other relevant information. classes: A list of class names corresponding to the class ids in the Detections object. filename: The name of the image file associated with the detections. image_shape: The shape of the image file associated with the detections. min_image_area_percentage: Minimum detection area relative to area of image associated with it. max_image_area_percentage: Maximum detection area relative to area of image associated with it. approximation_percentage: The percentage of polygon points to be removed from the input polygon, in the range [0, 1). Returns: An XML string in Pascal VOC format representing the detections. Note: ``detections`` is never mutated by this function; the source ``xyxy`` array is unchanged after the call. The function is therefore safe to call multiple times on the same ``Detections`` object. annotationfolderVOCfilenamesourcedatabasez roboflow.aisizewidthheightdepth segmented0z7Detections must include class_id for Pascal VOC export.zBDetections class_id must be an integer for Pascal VOC export, got .)maskmin_image_area_percentagemax_image_area_percentageapproximation_percentage)r)r"rr)r"rz )indent)rrrr ValueError isinstancer npintegertyper r r+appendr r toprettyxml) detectionsclassesr1 image_shaper<r=r>r6r5r7r.r/ file_namer2r3r4whdr8r"r;_class_idrpolygonsr next_object xml_strings r*detections_to_pascal_vocrSTsD'FE5&J H -FFK:z2IIN H -F&*-H!HM j& )D4!A ZAF4"A [AF4!A ZAF:{3IIN*4+%dAxA  VW W(S"**$56H~(+ x   5*C*C)A H $ /&w72D' !!+.  //DtDK   k *1+6[*!56BB$BOPJ r,c<t|gdDcgc] }t|}}g}i}|D]}t|j}tj j ||d} tj j| stj||<rt| } | j} tj|} | td|| jd| jdf} t!| || |\}}|||<|||fScc}w)a Loads PASCAL VOC XML annotations and returns the image name, a Detections instance, and a list of class names. Args: images_directory_path: The path to the directory containing the images. annotations_directory_path: The path to the directory containing the PASCAL VOC annotation files. force_masks: If True, forces masks to be loaded for all annotations, regardless of whether they are present. Returns: A tuple with a list of class names, a list of paths to images, and a dictionary with image paths as keys and corresponding Detections instances as values. )jpgjpegpng) directory extensionsz.xmlz Could not read image from path: rr)rrrstemospathjoinexistsr emptyrgetrootcv2imreadr@shapedetections_from_xml_obj)images_directory_pathannotations_directory_path force_masksr\ image_pathsrHr image_path image_stemannotation_pathtreer#image resolution_whr.s r*load_pascal_voc_annotationsros+0/+8N   D KGK!- *%** '',,'Aj\QUCVWww~~o.&0&6&6&8K # _%||~ :& =? |LM MQQ8 5 '=+  G#- J#-& K ,,;sDcg}g}g}|xs!td|jdD}|dd}|jdD]*} t| d} |j| | j d} | t dt t| d} t t| d} t t| d }t t| d }|j| | ||gtj|d |d ft }| jdD]7}t|}|d z}t||}||jtz}9|s|j|-|r&tj|tj }n%tjdtj }|d z}t!|D]}||vs|j|tj|D cgc]} |j#| c} }t%||rtj|t nd|}||fScc} w)aB Converts an XML object in Pascal VOC format to a Detections object. Expected XML format: ... dog 48 240 195 371 48 240 195 240 195 371 48 371 Returns: A tuple containing a Detections object and an updated list of class names, extended with the class names from the XML object. c32K|]}t|ywN)_with_poly_mask).0objs r* z*detections_from_xml_obj..s$!$$srNrrz(Missing bndbox in Pascal VOC annotation.rrrrrrdtyper)rrn)r)r"r;rO)anyfindall_get_required_textrEfindr@r rBzerosboolparse_polygon_pointsr astypearrayfloat32r_setr&r )r#rHrnrgr" class_namesmasks with_masksextended_classesru class_namebboxx1y1x2y2 object_maskpolygon_elementrmask_from_polygonxyxy_arrkrOr.s r*rdrdsDJDK)+E$(, X(>$!Jqz||H%&'V4 :&xx! <GH H #D&1 2 #D&1 2 #D&1 2 #D&1 2 RR$%-/XX 1 }Q/ 0.  #{{95 :O*?;G qLG /+!  ,33D9 9K :  LL %;&@ 88D 388F"**5 MH  ' $ $  # #A &'xx>IJ    +JH ,6RXXe4 (DJ ' '' Ks;I c(|jdduS)Nr)r})rus r*rsrsOs 88I d **r,c Bg}|jdD]=}|j td|jt |j?t j tdt|dDcgc]}||||dzfc}tScc}w)Nz.//*z/Missing polygon coordinate value in Pascal VOC.rrrrw) r{rr@rEr rBrrangelen)r coordinatescoordis r*rrSsK(, :: NO O3uzz?+, 887rrr)F)rerrfrrgrrz2tuple[list[str], list[str], dict[str, Detections]]) r#rrHrrnztuple[int, int]rgrrztuple[Detections, list[str]])rurrr)rrrznpt.NDArray[np.int_])rrrrrr)# __future__rr[pathlibrxml.etree.ElementTreerrranumpyrB numpy.typingtypingnptdefusedxml.ElementTreerrdefusedxml.minidomr supervision.dataset.utilsr supervision.detection.corer &supervision.detection.utils.convertersr r supervision.utils.filerr+rSrordrsrr|r,r*rs>" 5 2*D1S= .2> > >+> >L(+'*&*^^ ^^& ^ % ^ % ^$^ ^H3-3- #3-3-8 3-t a( a( a(#a( a( " a(H+ r,