djdZddlZddlmZej ZdejdejfdZdedede fdZ y) zCosine similarity utilities.N)embedding_resultuvct|dkr tdtjj |}tjj |}|dks|dkr td|j |||zz S)z2Computes cosine similarity between two embeddings.rz5Cannot compute cosing similarity on empty embeddings.z:Cannot compute cosine similarity on embedding with 0 norm.)len ValueErrornplinalgnormdot)rrnorm_unorm_vs |/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/mediapipe/tasks/python/components/utils/cosine_similarity.py_compute_cosine_similarityrsw Vq[ L MM 99>>! & 99>>! & q[FaK D FF qVf_ %%returnct|jt|jk7r8tdt|jdt|jd|jjtk(r=|jjtk(r t |j|jS|jjt jk(r|jjt jk(r\t |jjdjd|jjdjdStd)aFUtility function to compute cosine similarity between two embedding. May return an InvalidArgumentError if e.g. the feature vectors are of different types (quantized vs. float), have different sizes, or have an L2-norm of 0. Args: u: An embedding. v: An embedding. Returns: Cosine similarity value. zHCannot compute cosine similarity between embeddings of different sizes (z vs. z).int8floatzHCannot compute cosine similarity between quantized and float embeddings.) r embeddingrdtyperrr uint8viewastype)rrs rcosine_similarityr's Q[[)) Q[[)*%AKK0@/AE FF[[%AKK$5$5$> %akk1;; ??[["(("q{{'8'8BHH'D %  ''0  ''0  ' ((r) __doc__numpyr ,mediapipe.tasks.python.components.containersr Embedding _Embeddingndarrayrrrrrr#sP#I  ' '  &"** & & (( (u(r