mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-03-13 04:45:32 +08:00
SikongJueluo
4da08dc3d3
- Remove dino_compressor.py and segament_compressor.py - Rewrite pipeline.py to inline DINO into HashPipeline - Maintain backward compatibility: SAMHashPipeline alias - Update tests and benchmark.py
20 lines
585 B
Python
20 lines
585 B
Python
from .common import BinarySign, bits_to_hash, hamming_distance, hamming_similarity, hash_to_bits
|
|
from .hash_compressor import HashCompressor, HashLoss, VideoPositiveMask
|
|
from .pipeline import HashPipeline, SAMHashPipeline, create_pipeline_from_config
|
|
from .train import train
|
|
|
|
__all__ = [
|
|
"train",
|
|
"HashCompressor",
|
|
"HashLoss",
|
|
"VideoPositiveMask",
|
|
"HashPipeline",
|
|
"SAMHashPipeline", # Backward compatibility alias
|
|
"create_pipeline_from_config",
|
|
"BinarySign",
|
|
"hamming_distance",
|
|
"hamming_similarity",
|
|
"bits_to_hash",
|
|
"hash_to_bits",
|
|
]
|