feat(utils): add feature extraction utilities and tests

This commit is contained in:
2026-03-05 20:48:53 +08:00
parent a16b376dd7
commit 5be4709acf
13 changed files with 247 additions and 138 deletions

View File

@@ -1,3 +1,14 @@
from .common import get_device, get_output_diretory
from .feature_extractor import (
extract_batch_features,
extract_single_image_feature,
infer_vector_dim,
)
__all__ = ["get_device", "get_output_diretory"]
__all__ = [
"get_device",
"get_output_diretory",
"infer_vector_dim",
"extract_single_image_feature",
"extract_batch_features",
]