dj dZddlZddlmZmZddlZddlmZddl m Z ejGddZ ejGdd Z y) zEmbeddings data class.N)ListOptional)embedding_result_c) doc_controlscXeZdZUdZej ed<dZee ed<dZ ee ed<y) EmbeddingaEmbedding result for a given embedder head. Attributes: embedding: The actual embedding, either floating-point or scalar-quantized. head_index: The index of the embedder head that produced this embedding. This is useful for multi-head models. head_name: The name of the embedder head, which is the corresponding tensor metadata name (if any). This is useful for multi-head models. embeddingN head_index head_name) __name__ __module__ __qualname____doc__npndarray__annotations__r rintr str/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/mediapipe/tasks/python/components/containers/embedding_result.pyrrs/ZZ"*hsm"!)Xc]!rrceZdZUdZeeed<dZee ed<e e jde jddfdZy)EmbeddingResulta%Embedding results for a given embedder model. Attributes: embeddings: A list of `Embedding` objects. timestamp_ms: The optional timestamp (in milliseconds) of the start of the chunk of data corresponding to these results. This is only used for embedding extraction on time series (e.g. audio embedding). 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. embeddingsN timestamp_msc_resultreturncg}t|jD]}|j|}d}|jr=t |jd|j }t j|t}|jrGt |jd|j }t j|t j}t||j|jr|jjdnd}|j||j r |j"nd}t%||S)z9Converts a C EmbeddingResult to a Python EmbeddingResult.N)dtypezutf-8)r r r )rr)rangeembeddings_countrfloat_embeddinglist values_countrarrayfloatquantized_embeddinguint8rr r decodeappendhas_timestamp_msrr) clsrpython_embeddingsi c_embeddingr embedding_arraypython_embeddingrs r from_ctypeszEmbeddingResult.from_ctypes;s*  8,, -1''*ki  $ $  ' '(B+*B*B C HH_E:  ( (  + +,Fk.F.F G HH_BHH= " ++&&##**73 /0/10-5,E,E8((4L $< r)r r rrrrrrrr classmethodrdo_not_generate_docsrEmbeddingResultCr2rrrrr*sW 9o $, $$$ '88  % rr)r dataclassestypingrrnumpyr,mediapipe.tasks.python.components.containersr1mediapipe.tasks.python.core.optional_dependenciesr dataclassrrrrrr<sW!KJ  " " "  222r