cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated embeddings container to use a timestamp and made parameters - head_index,head_name optional

Browse Source
This commit is contained in:
kinaryml 2022-11-08 23:51:55 -08:00
parent d327b3649d
commit dc30cf9732
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
mediapipe/tasks/python/components/containers/embeddings.py
View File

@ -110,8 +110,8 @@ class Embedding:
""" """
embedding: np.ndarray embedding: np.ndarray
head_index: int head_index: Optional[int] = None
head_name: str head_name: Optional[str] = None
@doc_controls.do_not_generate_docs @doc_controls.do_not_generate_docs
def to_pb2(self) -> _EmbeddingProto: def to_pb2(self) -> _EmbeddingProto:
@ -167,9 +167,16 @@ class EmbeddingResult:
"""Embedding results for a given embedder model. """Embedding results for a given embedder model.
Attributes: Attributes:
embeddings: A list of `Embedding` objects. 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.
""" """
embeddings: List[Embedding] embeddings: List[Embedding]
timestamp_ms: Optional[int] = None
@doc_controls.do_not_generate_docs @doc_controls.do_not_generate_docs
def to_pb2(self) -> _EmbeddingResultProto: def to_pb2(self) -> _EmbeddingResultProto: