SikongJueluo/Mini-Nav
1
0
Fork 0
mirror of https://github.com/SikongJueluo/Mini-Nav.git synced 2026-03-12 12:25:32 +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

22
mini-nav/database.py
View File

@@ -4,14 +4,16 @@ import lancedb
import pyarrow as pa
from configs import cfg_manager
db_schema = pa.schema(
[
pa.field("id", pa.int32()),
pa.field("label", pa.string()),
pa.field("vector", pa.list_(pa.float32(), 1024)),
pa.field("binary", pa.binary()),
]
)
def _build_database_schema():
return pa.schema(
[
pa.field("id", pa.int32()),
pa.field("label", pa.string()),
pa.field("vector", pa.list_(pa.float32(), 1024)),
pa.field("binary", pa.binary()),
]
)
class DatabaseManager:
@@ -34,7 +36,9 @@ class DatabaseManager:
# 初始化数据库与表格
self.db = lancedb.connect(db_path)
if "default" not in self.db.list_tables().tables:
self.table = self.db.create_table("default", schema=db_schema)
self.table = self.db.create_table(
"default", schema=_build_database_schema()
)
else:
self.table = self.db.open_table("default")