
    Ng                     Z     G d  de           Z G d de           Z G d de           ZdS )c                   b    e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdS )InputDataFieldsaS  Names for the input tensors.
    Holds the standard data field names to use for identifying input tensors. This
    should be used by the decoder to identify keys for the returned tensor_dict
    containing input tensors. And it should be used by the model to identify the
    tensors it needs.
    Attributes:
        image: image.
        image_additional_channels: additional channels.
        key: unique key corresponding to image.
        filename: original filename of the dataset (without common path).
        gt_image_classes: image-level class labels.
        gt_image_confidences: image-level class confidences.
        gt_labeled_classes: image-level annotation that indicates the
            classes for which an image has been labeled.
        gt_boxes: coordinates of the ground truth boxes in the image.
        gt_classes: box-level class labels.
        gt_confidences: box-level class confidences. The shape should be
            the same as the shape of gt_classes.
        gt_label_types: box-level label types (e.g. explicit negative).
        gt_is_crowd: [DEPRECATED, use gt_group_of instead]
            is the groundtruth a single object or a crowd.
        gt_area: area of a groundtruth segment.
        gt_difficult: is a `difficult` object
        gt_group_of: is a `group_of` objects, e.g. multiple objects of the
            same class, forming a connected group, where instances are heavily
            occluding each other.
        gt_instance_masks: ground truth instance masks.
        gt_instance_boundaries: ground truth instance boundaries.
        gt_instance_classes: instance mask-level class labels.
        gt_label_weights: groundtruth label weights.
        gt_weights: groundtruth weight factor for bounding boxes.
        image_height: height of images, used to decode
        image_width: width of images, used to decode
    imageimage_idfilenamebboxclsconfidenceslabel_typesimg_clsimg_confidenceslabeled_clsis_crowdarea	difficultgroup_ofinstance_masksinstance_boundariesinstance_classes
img_height	img_widthimg_sizeN)__name__
__module____qualname____doc__r   keyr   gt_boxes
gt_classesgt_confidencesgt_label_typesgt_image_classesgt_image_confidencesgt_labeled_classesgt_is_crowdgt_areagt_difficultgt_group_ofgt_instance_masksgt_instance_boundariesgt_instance_classesimage_heightimage_width
image_size     T/var/www/html/ai-engine/env/lib/python3.11/site-packages/effdet/evaluation/fields.pyr   r      s        ! !D E
CHHJ"N"N ,&KGLK(2,LKJJJr/   r   c                   &    e Zd ZdZdZdZdZdZdZdS )DetectionResultFieldsap  Naming conventions for storing the output of the detector.
    Attributes:
        source_id: source of the original image.
        key: unique key corresponding to image.
        detection_boxes: coordinates of the detection boxes in the image.
        detection_scores: detection scores for the detection boxes in the image.
        detection_multiclass_scores: class score distribution (including background)
            for detection boxes in the image including background class.
        detection_classes: detection-level class labels.
        detection_masks: contains a segmentation mask for each detection box.
    r   r   scorer   masksN)	r   r   r   r   r   detection_boxesdetection_scoresdetection_classesdetection_masksr.   r/   r0   r2   r2   ;   s5        
 
 COOOOr/   r2   c                   F    e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdS )BoxListFieldsa  Naming conventions for BoxLists.
    Attributes:
        boxes: bounding box coordinates.
        classes: classes per bounding box.
        scores: scores per bounding box.
        weights: sample weights per bounding box.
        objectness: objectness score per bounding box.
        masks: masks per bounding box.
        boundaries: boundaries per bounding box.
        keypoints: keypoints per bounding box.
        keypoint_heatmaps: keypoint heatmaps per bounding box.
        is_crowd: is_crowd annotation per bounding box.
    boxesclassesscoresweightsr	   
objectnessr4   
boundaries	keypointskeypoint_visibilitieskeypoint_heatmapsr   r   N)r   r   r   r   r;   r<   r=   r>   r	   r?   r4   r@   rA   rB   rC   r   r   r.   r/   r0   r:   r:   O   s]          EGFGKJEJI3+HHHHr/   r:   N)objectr   r2   r:   r.   r/   r0   <module>rE      s   6 6 6 6 6f 6 6 6r    F   (    F     r/   