^j$ddlmZddlZddlmZmZddlZddlmZ ddl m Z ddl m Z ddlmZ ddZ ddZ dd Z dd Z dd Z dd Zdd Z d ddZy)) annotationsN)Anycast)Image)CLASS_NAME_DATA_FIELD) mask_to_xyxycr|djjjjt}t ||i}t |djjj|djjj||S)a[ Process the result of Transformers object detection functions such as `post_process` (v4) and `post_process_detection` (v5). Args: detection_result: Dictionary containing detection results with keys 'boxes', 'labels', and 'scores'. id2label: A dictionary mapping class IDs to labels, typically part of the `transformers` model configuration. If provided, the resulting dictionary will include class names. Returns: Processed detection result including bounding boxes, confidence scores, class IDs, and data. labelsboxesscores)xyxy confidenceclass_iddata)cpudetachnumpyastypeintappend_class_names_to_datadict)detection_resultid2label class_idsrs s/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/supervision/detection/tools/transformers.py%process_transformers_detection_resultrs$!*..0779??AHHMI %i2 >D  g & * * , 3 3 5 ; ; =#H-113::<BBD   c pd|vr t||Sd}d|vr/|djjj}|djjjj t }|djjjj t }t||n t||tj|dn||djjj|t||i S) a Process the result of Transformers segmentation functions such as `post_process_panoptic`, `post_process_segmentation`, and `post_process_instance` (v4). Args: segmentation_result: Dictionary containing segmentation results with keys 'masks', 'labels', and 'scores'. id2label: A dictionary mapping class IDs to labels, typically part of the `transformers` model configuration. If provided, the resulting dictionary will include class names. Returns: Processed segmentation result including bounding boxes, masks, confidence scores, class IDs, and data. png_stringNr masksr axisr r maskrrr) 4process_transformers_v4_panoptic_segmentation_resultrrrrboolrrrnpsqueezer)segmentation_resultrr r rs r+process_transformers_v4_segmentation_resultr++s&**C    ) )'0446==?EEGE#G,00299;AACJJ4P'1557>>@FFHOOPST +e1D.3.?E*U*8488:AACIIK+IxD   rc|jjdk(r8|jjj }t ||St ttttf||S)a Process the result of Transformers segmentation functions such as `post_process_semantic_segmentation`, `post_process_instance_segmentation`, and `post_process_panoptic_segmentation` (v5). Args: segmentation_result: Either a dictionary containing segmentation results (`segments_info` and `segmentation`) or a tensor object representing a panoptic segmentation map. id2label: A dictionary mapping class IDs to labels, typically part of the `transformers` model configuration. If provided, the resulting dictionary will include class names. Returns: Processed segmentation result including bounding boxes, masks, confidence scores, class IDs, and data. Tensor) __class____name__rrr4process_transformers_v5_panoptic_segmentation_result@process_transformers_v5_semantic_or_instance_segmentation_resultrrstrr)r*rsegmentation_arrays r+process_transformers_v5_segmentation_resultr4Rsp($$--90446==?EEGC    L T#s(^018 rc|d}tj|Dcgc]}|d c}}tj|Dcgc]}|d c}}|djjj }tj|Dcgc] }||dk( c}j t }t||i}tt|||||Scc}wcc}wcc}w)a Process the result of Transformers segmentation functions such as `post_process_semantic_segmentation` and `post_process_instance_segmentation` (v5). Args: segmentation_result: Dictionary containing segmentation results with keys `segments_info` and `segmentation`. id2label: A dictionary mapping class IDs to labels, typically part of the `transformers` model configuration. If provided, the resulting dictionary will include class names. Returns: Processed segmentation result including bounding boxes, masks, confidence scores, class IDs, and data. segments_infoscorelabel_id segmentationidr$) r(arrayrrrrr'rrr) r*rr6segmentr rr3r rs rr1r1ps$(8M XX}EGww'E FF]K''*-KLI,^<@@BIIKQQS HHD  %    FK Ks C# C(C-cP|d}|d}tj|Dcgc]}|d c}}t|}tj|Dcgc] }||dk( c}jt}t ||i}t t||||Scc}wcc}w)a. Process the result of the Transformers function `post_process_panoptic` (v4). Args: segmentation_result: Dictionary containing segmentation results with keys such as 'png_string' and 'segments_info'. id2label: A dictionary mapping class IDs to labels, typically part of the `transformers` model configuration. If provided, the resulting dictionary will include class names. Returns: Processed segmentation result including bounding boxes, masks, class IDs, and data. r6r category_id)rr:r r%rr)r(r; png_string_to_segmentation_arrayrr'rrr) r*rr6rr<rr3r rs rr&r&s"(8M$\2J NW'-0NOI9ZP HHD  %    O Ks BB#ctj|}tj|Dcgc]}||k( c}djt}t ||i}t t||||Scc}w)a Process the result of the Transformers function `post_process_panoptic_segmentation` (v5). Args: segmentation_array: Segmentation array. id2label: A dictionary mapping class IDs to labels, typically part of the `transformers` model configuration. If provided, the resulting dictionary will include class names. Returns: Processed segmentation result including bounding boxes, masks, class IDs, and data. rr"r?)r(uniquestackrr'rrr)r3rrrr rs rr0r0so" ,-I HH8ABH x 'B  fTl  &i2 >D \%(uyt TT Cs A6ctjtj|}t j |tj }ttjt|dddddfS)aV Convert a PNG byte string to a label mask array. Args: png_string: A byte string representing the PNG image. Returns: A label mask array with shape (H, W), where H and W are the height and width of the image. Each unique value in the array represents a different object or category. )dtypeNr) ropenioBytesIOr(r;uint8rnptNDArrayr)rimager%s rr@r@sO JJrzz*- .E 88E *D  C $q!Qw- 00rcz|i}|/tj|Dcgc]}|| c}}||t<|Scc}w)a Helper function to create or append to a data dictionary with class names if available. Args: class_ids: Array of class IDs. id2label: A dictionary mapping class IDs to labels, typically part of the `transformers` model configuration. If provided, the resulting dictionary will include class names. data: An existing data dictionary to append to. Returns: Dictionary containing class names if id2label is provided. )r(r;r)rrrr class_namess rrrsH& |hh9Mx 2MN &1 "# K Ns 8)rdict[str, Any]rdict[int, str] | NonereturnrO)r*rOrrPrQrO)r*rrrPrQrO)r3npt.NDArray[Any]rrPrQrO)rbytesrQrR)N)rrRrrPrzdict[str, Any] | NonerQrO) __future__rrGtypingrrrr( numpy.typingrJPILrsupervision.configr&supervision.detection.utils.convertersrrr+r4r1r&r0r@rrrr[s" 4?$0E:$ '$ 3H$ $ N(=