khzldZddlZddlmZmZmZej jGddeZ y)aBounding Box List definition. BoxList represents a list of bounding boxes as tensorflow tensors, where each bounding box is represented as a row of 4 numbers, [y_min, x_min, y_max, x_max]. It is assumed that all bounding boxes within a given list correspond to a single image. See also box_list.py for common box related operations (such as area, iou, etc). Optionally, users can add additional related fields (such as weights). We assume the following things to be true about fields: * they correspond to boxes in the box_list along the 0th dimension * they have inferable rank at graph construction time * all dimensions except for possibly the 0th can be inferred (i.e., not None) at graph construction time. Some other notes: * Following tensorflow conventions, we use height, width ordering, and correspondingly, y,x (or ymin, xmin, ymax, xmax) ordering * Tensors are always provided as (flat) [N, 4] tensors. N)OptionalListDictceZdZUdZeeejfed<dZ dZ dZ dZ dedejfd Z defd Zd Zd Zdefd ZdedejfdZdZdZddeeefdZdZy)BoxListzBox collection.datact|jdk7s|jddk7r td|jtj k7r tdd|i|_y)zConstructs box collection. Args: boxes: a tensor of shape [N, 4] representing box corners Raises: ValueError: if invalid dimensions for bbox data or if bbox data is not in float32 format.  Invalid dimensions for box data.z)Invalid tensor type: should be tf.float32boxesN)lenshape ValueErrordtypetorchfloat32rselfrs \/var/www/teggl/fontify/venv/lib/python3.12/site-packages/effdet/object_detection/box_list.py__init__zBoxList.__init__,sX u{{ q EKKOq$8?@ @ ;;%-- 'HI Ie$ c:|jdjdS)zReturns number of boxes held in collection. Returns: a tensor representing the number of boxes held in the collection. rr)rrrs r num_boxeszBoxList.num_boxes;s yy!''**rc6|jjS)zReturns all fields.)rkeysrs rget_all_fieldszBoxList.get_all_fieldsCsyy~~rcvg}|jjD]}|dk7s |j||S)z@Returns all non-box fields (i.e., everything not named 'boxes').r)rrappend)rextraks rget_extra_fieldszBoxList.get_extra_fieldsGs<! AG| Q  rfield field_datac"||j|<y)aAdd field to box list. This method can be used to add related box data such as weights/labels, etc. Args: field: a string key to access the data via `get` field_data: a tensor containing the data to store in the BoxList Nr)rr%r&s r add_fieldzBoxList.add_fieldPs& %rc||jvSNr(rr%s r has_fieldzBoxList.has_field[s !!rc$|jdS)zConvenience function for accessing box coordinates. Returns: a tensor with shape [N, 4] representing box coordinates. r) get_fieldrs rrz BoxList.boxes_s ~~g&&rct|jdk7s|jddk7r td||jd<y)zConvenience function for setting box coordinates. Args: boxes: a tensor of shape [N, 4] representing box corners Raises: ValueError: if invalid dimensions for bbox data r r r r rN)rrrrrs r set_boxeszBoxList.set_boxeshs> u{{ q EKKOq$8?@ @" 'rc`|j|std|d|j|S)aAccesses a box collection and associated fields. This function returns specified field with object; if no field is specified, it returns the box coordinates. Args: field: this optional string parameter can be used to specify a related field to be accessed. Returns: a tensor representing the box collection or an associated field. Raises: ValueError: if invalid field field  does not existr-rrr,s rr/zBoxList.get_fieldus3~~e$veWO<= =yyrvaluecb|j|std|d||j|<y)a8Sets the value of a field. Updates the field of a box_list with a given value. Args: field: (string) name of the field to set value. value: the value to assign to the field. Raises: ValueError: if the box_list does not have specified field. r3r4Nr5)rr%r6s r set_fieldzBoxList.set_fields2~~e$veWO<= =  %rc|j}|jj\}}}}||z }||z }||dz z}||dz z} || ||gS)zComputes the center coordinates, height and width of the boxes. Returns: a list of 4 1-D tensors [ycenter, xcenter, height, width]. g@)rtunbind) r box_cornersyminxminymaxxmaxwidthheightycenterxcenters r get_center_coordinates_and_sizesz(BoxList.get_center_coordinates_and_sizessi jjl !,!7!7!9dD$t "$#&%00rc|jjdd\}}}}|jtj||||gdy)z?Transpose the coordinate representation in a boxlist. r )dimN)rchunkr1rcat)ry_minx_miny_maxx_maxs rtranspose_coordinateszBoxList.transpose_coordinatessH&*ZZ\%7%7q%7%A"ueU uyy%u!=qABrNfieldsci}||j}|D]2}|j|s td|j|||<4|S)aRetrieves specified fields as a dictionary of tensors. Args: fields: (optional) list of fields to return in the dictionary. If None (default), all fields are returned. Returns: tensor_dict: A dictionary of tensors specified by fields. Raises: ValueError: if specified field is not contained in boxlist. z)boxlist must contain all specified fields)rr-rr/)rrP tensor_dictr%s ras_tensor_dictzBoxList.as_tensor_dicts^ >((*F 7E>>%( !LMM!%!6K  7rc4|jdjS)Nr)rdevicers rrUzBoxList.devicesyy!(((rr+)__name__ __module__ __qualname____doc__rstrrTensor__annotations__rrrr$r)r-rr1r/r8rErOrrrSrUrrrr's sELL !! %+  &s & &"s"' # s &!s!5<<! 1CXd3i%8.)rr) rYrtypingrrrjitscriptobjectrr]rrrbs=( ''])f])])r