o
e% @ s~ d dl Z d dlmZ d dlmZmZmZmZ d dlm Z d dl
m Z m Z m
Z
mZ d dlZd dlZd dlmZ d dlmZ d dlmZmZ d d lmZ d'd
ejd e d eej d
e fddZ d(dedee de dee e e f de de de d
e fddZ d)d e d e d e!d
eee ee ejf ee ef f fd d!Z" d)d"e dee d e!d
eeee f fd#d$Z#d e d
ejfd%d&Z$dS )* N)Path)DictListOptionalTuple) parseString)Element
SubElementparsetostring) approximate_mask_with_polygons)
Detections)polygon_to_maskpolygon_to_xyxy)list_files_with_extensionsxyxynamepolygonreturnc C s
t d}t|d}||_| d7 } t|d}t|d}tt| d |_t|d}tt| d |_t|d}tt| d |_t|d
} tt| d | _|d ur|d7 }t|d }
t|dd
D ]"\} } | \}
}t|
d| }t|
|_t|
d| }t||_q`|S )Nobjectr bndboxxminr yminxmax ymax r )startxy)r r textstrint enumerate)r r r root object_namer r r r r object_polygonindexpoint x_coordinate y_coordinater r r, U/var/www/myenv/lib/python3.10/site-packages/supervision/dataset/formats/pascal_voc.pyobject_to_pascal_voc s.
r. ? ?
detectionsclassesfilename image_shapemin_image_area_percentagemax_image_area_percentageapproximation_percentagec C s2 |\}}} t d}
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 ]9\}}}}}|| }|durt||||d}|D ]}t|d}t|||d}|
| qlqSt||d}|
| qStt |
j
dd} | S )a
Converts Detections object to Pascal VOC XML format.
Args:
detections (Detections): A Detections object containing bounding boxes,
class ids, and other relevant information.
classes (List[str]): A list of class names corresponding to the
class ids in the Detections object.
filename (str): The name of the image file associated with the detections.
image_shape (Tuple[int, int, int]): The shape of the image
file associated with the detections.
min_image_area_percentage (float): Minimum detection area
relative to area of image associated with it.
max_image_area_percentage (float): Maximum detection area
relative to area of image associated with it.
approximation_percentage (float): The percentage of
polygon points to be removed from the input polygon, in the range [0, 1).
Returns:
str: An XML string in Pascal VOC format representing the detections.
annotationfolderVOCr4 sourcedatabasez roboflow.aisizewidthheightdepth segmented0N)maskr6 r7 r8 r )r r r )r r z )indent) r r r! r" r r r. appendr r toprettyxml) r2 r3 r4 r5 r6 r7 r8 r@ r? rA r9 r: file_namer< r= r> whdrB r rD _class_idr polygonsr next_object
xml_stringr, r, r- detections_to_pascal_voc3 sL
rR Fimages_directory_pathannotations_directory_path force_masksc C s t | g dd}g }i }i }|D ]N}t|j}t|}t|} tj|| d}
tj |
s;| ||< t
||< qt |
} |
} | jd | jd f}
t| ||
|\}}| ||< |||< q|||fS )a7
Loads PASCAL VOC XML annotations and returns the image name,
a Detections instance, and a list of class names.
Args:
images_directory_path (str): The path to the directory containing the images.
annotations_directory_path (str): The path to the directory containing the
PASCAL VOC annotation files.
force_masks (bool, optional): If True, forces masks to be loaded for all
annotations, regardless of whether they are present.
Returns:
Tuple[List[str], Dict[str, np.ndarray], Dict[str, Detections]]: A tuple
containing a list of class names,
a dictionary with image names as keys and
images as values, and a dictionary with image names as
keys and corresponding Detections instances as values.
)jpgjpegpng) directory
extensionsz.xmlr r )r r stemr" cv2imreadospathjoinexistsr
emptyr
getrootshapedetections_from_xml_obj)rS rT rU image_pathsr3 images annotations
image_path
image_nameimageannotation_pathtreer%
resolution_whr9 r, r, r- load_pascal_voc_annotations s0
ro r% c s g }g }g }d}|dd | dD ]c}|dj} || |d}
t|
dj} t|
dj} t|
dj}
t|
d j}|| | |
|g |d
du}|rY|n|}| d
D ]}t|}|d 8 }t||d }|| q`qt|d
krt |nt
d}|d 8 }t |D ] }| vr | qt fdd|D }|rt |t |
t|d}| fS t ||d}| fS )a`
Converts an XML object in Pascal VOC format to a Detections object.
Expected XML format:
...
Returns:
Tuple[Detections, List[str]]: A tuple containing a Detections object and an
updated list of class names, extended with the class names
from the XML object.
FNr r r r r r r r r )r rn r )r c s g | ]} |qS r, )r( ).0
class_nameextended_classesr, r-
z+detections_from_xml_obj..)r rD rN )r rN )findallfindr! rG r# parse_polygon_pointsr lennparrayrb setr
astypebool)r% r3 rn rU r class_namesmasks
with_masksobjrr bboxx1y1x2y2r mask_from_polygonkrN r9 r, rs r- re sR "
re c s8 dd | dD t fddtdt dD S )Nc S s g | ]}t |jqS r, )r# r! )rq coordr, r, r- ru rv z(parse_polygon_points..z.//*c s g | ] } | |d fqS )r r, )rq i coordinatesr, r- ru s r r )rw r{ r| rangerz rE r, r r- ry s ry )N)r/ r0 r1 )F)%r^ pathlibr typingr r r r xml.dom.minidomr xml.etree.ElementTreer r r
r r\ numpyr{ supervision.dataset.utilsr supervision.detection.corer
supervision.detection.utilsr r supervision.utils.filer ndarrayr" r. r# floatrR r ro re ry r, r, r, r- s|
(
Y"
:
V