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

Fixed a bug in embedding_result

Browse Source
This commit is contained in:
kinaryml 2022-11-12 08:55:56 -08:00
parent a3788a23bc
commit a7ed160a8e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/tasks/python/components/containers/embedding_result.py
View File

@ -51,7 +51,7 @@ class Embedding:
bytearray(pb2_obj.quantized_embedding.values)) bytearray(pb2_obj.quantized_embedding.values))
float_embedding = np.array(pb2_obj.float_embedding.values, dtype=float) float_embedding = np.array(pb2_obj.float_embedding.values, dtype=float)
if not quantized_embedding: if not pb2_obj.quantized_embedding.values:
return Embedding( return Embedding(
embedding=float_embedding, embedding=float_embedding,
head_index=pb2_obj.head_index, head_index=pb2_obj.head_index,