SikongJueluo/Mini-Nav
1
0
Fork 0
mirror of https://github.com/SikongJueluo/Mini-Nav.git synced 2026-03-12 20:35:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(utils): add feature extraction utilities and tests

Browse Source
This commit is contained in:
SikongJueluo
2026-03-05 20:48:53 +08:00
parent a16b376dd7
commit 5be4709acf
13 changed files with 247 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mini-nav/benchmarks/runner.py
View File

@@ -149,9 +149,9 @@ def run_benchmark(
sample = train_dataset[0]
sample_image = sample["img"]
from .tasks.retrieval import _infer_vector_dim
from utils.feature_extractor import infer_vector_dim
vector_dim = _infer_vector_dim(processor, model, sample_image)
vector_dim = infer_vector_dim(processor, model, sample_image)
print(f"Model output dimension: {vector_dim}")
# Ensure table exists with correct schema