mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-03-12 20:35:31 +08:00
feat(utils): add feature extraction utilities and tests
This commit is contained in:
@@ -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",
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user