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:
63
mini-nav/feature_compressor/examples/basic_usage.py
Normal file
63
mini-nav/feature_compressor/examples/basic_usage.py
Normal file
@@ -0,0 +1,63 @@
|
||||
"""Basic usage example for DINOv2 Feature Compressor."""
|
||||
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
# Add parent to path for imports
|
||||
sys.path.insert(0, str(Path(__file__).parent.parent.parent))
|
||||
|
||||
import requests
|
||||
from PIL import Image
|
||||
import io
|
||||
|
||||
from dino_feature_compressor import DINOv2FeatureExtractor, FeatureVisualizer
|
||||
|
||||
|
||||
def main():
|
||||
# Initialize extractor
|
||||
print("Initializing DINOv2FeatureExtractor...")
|
||||
extractor = DINOv2FeatureExtractor()
|
||||
|
||||
# Download and save test image
|
||||
print("Downloading test image...")
|
||||
url = "http://images.cocodataset.org/val2017/000000039769.jpg"
|
||||
response = requests.get(url)
|
||||
img = Image.open(io.BytesIO(response.content))
|
||||
|
||||
test_image_path = "/tmp/test_image.jpg"
|
||||
img.save(test_image_path)
|
||||
print(f"Image saved to {test_image_path}")
|
||||
|
||||
# Extract features
|
||||
print("Extracting features...")
|
||||
result = extractor.process_image(test_image_path)
|
||||
|
||||
print(f"\n=== Feature Extraction Results ===")
|
||||
print(f"Original features shape: {result['original_features'].shape}")
|
||||
print(f"Compressed features shape: {result['compressed_features'].shape}")
|
||||
print(f"Processing time: {result['metadata']['processing_time']:.3f}s")
|
||||
print(f"Compression ratio: {result['metadata']['compression_ratio']:.2f}x")
|
||||
print(f"Feature norm: {result['metadata']['feature_norm']:.4f}")
|
||||
print(f"Device: {result['metadata']['device']}")
|
||||
|
||||
# Visualize
|
||||
print("\nGenerating visualization...")
|
||||
viz = FeatureVisualizer()
|
||||
|
||||
fig = viz.plot_histogram(
|
||||
result["compressed_features"], title="Compressed Features Distribution"
|
||||
)
|
||||
|
||||
output_path = (
|
||||
Path(__file__).parent.parent.parent / "outputs" / "basic_usage_histogram"
|
||||
)
|
||||
output_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
viz.save(fig, str(output_path), formats=["html"])
|
||||
print(f"Visualization saved to {output_path}.html")
|
||||
|
||||
print("\nDone!")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user