mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-03-12 12:25:32 +08:00
feat(feature-compressor): add DINOv2 feature extraction and compression pipeline
This commit is contained in:
20
mini-nav/configs/config.py
Normal file
20
mini-nav/configs/config.py
Normal file
@@ -0,0 +1,20 @@
|
||||
from enum import Enum
|
||||
from pathlib import Path
|
||||
from typing import Dict
|
||||
|
||||
import yaml
|
||||
|
||||
|
||||
class Config(Enum):
|
||||
FEATURE_COMPRESSOR = "feature_compressor.yaml"
|
||||
|
||||
|
||||
def get_config_dir() -> Path:
|
||||
return Path(__file__).parent
|
||||
|
||||
|
||||
def get_default_config(config_type: Config) -> Dict[Unknown, Unknown]:
|
||||
config_path = get_config_dir() / config_type.value
|
||||
|
||||
with open(config_path) as f:
|
||||
return yaml.safe_load(f)
|
||||
Reference in New Issue
Block a user