dj/dZddlZddlmZmZddlmZddlmZddl m Z ejGddZ ejGd d Z y) zClassifications data class.N)ListOptional)category)classification_result_c) doc_controlsceZdZUdZeej ed<eed<dZ e e ed<e e jdej ddfdZy) ClassificationsaRepresents the classification results for a given classifier head. Attributes: categories: The array of predicted categories, usually sorted by descending scores (e.g. from high to low probability). head_index: The index of the classifier head these categories refer to. This is useful for multi-head models. head_name: The name of the classifier head, which is the corresponding tensor metadata name. categories head_indexN head_namec_structreturnch|jr\|jdkDrMt|jDcgc].}tjj |j|0}}ng}t ||j|jr |jjdSdScc}w)z@Creates a `Classifications` object from the given ctypes struct.rzutf-8N)r r r ) r categories_countrange category_libCategory from_ctypesr r r decode)clsr ir s /root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/mediapipe/tasks/python/components/containers/classification_result.pyrzClassifications.from_ctypes)s x881<223    + +H,?,?,B Cj j &&2:2D2DH   % %g .  KO  s3B/)__name__ __module__ __qualname____doc__rrr__annotations__intr rstr classmethodrdo_not_generate_docsrClassificationsCrrr r sc <(())/!)Xc]!$$,==%r$r ceZdZUdZeeed<dZee ed<e e jde jddfdZy)ClassificationResultaUContains the classification results of a model. Attributes: classifications: A list of `Classifications` objects, each for a head of the model. timestamp_ms: The optional timestamp (in milliseconds) of the start of the chunk of data corresponding to these results. This is only used for classification on time series (e.g. audio classification). In these use cases, the amount of data to process might exceed the maximum size that the model can process: to solve this, the input data is split into multiple chunks starting at different timestamps. classificationsN timestamp_msr rc|jrR|jdkDrCt|jDcgc]$}tj |j|&}}ng}|j r |j nd}t||Scc}w)z=Creates a `ClassificationResult` object from a ctypes struct.rN)r'r()r'classifications_countrr rhas_timestamp_msr(r&)rr rr'r(s rrz ClassificationResult.from_ctypesRs H$B$BQ$F778  % %h&>&>q&A Bo o,4,E,E8((4L 'l s)B)rrrrrr rr(rrr rr!rClassificationResultCrr#r$rr&r&@sX (( $, $$$,BB%r$r&)r dataclassestypingrr,mediapipe.tasks.python.components.containersrrr1mediapipe.tasks.python.core.optional_dependenciesr dataclassr r&r#r$rr2sX"!QPJ $$$N """r$