SikongJueluo/Mini-Nav
1
0
Fork 0
mirror of https://github.com/SikongJueluo/Mini-Nav.git synced 2026-07-13 04:25:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(compressors): consolidate pipeline and improve mask handling

Browse Source
This commit is contained in:
SikongJueluo
2026-03-26 19:00:13 +08:00
parent 90d5a8f08a
commit 968819e113
11 changed files with 302 additions and 121 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
mini-nav/compressors/__init__.py
View File

@@ -1,6 +1,12 @@
from .common import BinarySign, bits_to_hash, hamming_distance, hamming_similarity, hash_to_bits
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 .pipeline import HashPipeline, create_pipeline_from_config
from .train import train
__all__ = [
@@ -9,7 +15,6 @@ __all__ = [
"HashLoss",
"VideoPositiveMask",
"HashPipeline",
"SAMHashPipeline", # Backward compatibility alias
"create_pipeline_from_config",
"BinarySign",
"hamming_distance",