feat(compressors): add object scoring and selection for SAM masks

This commit is contained in:
2026-03-30 16:31:06 +08:00
parent f6c1a67e88
commit a809803979
8 changed files with 581 additions and 5 deletions

View File

@@ -6,6 +6,14 @@ from .common import (
hash_to_bits,
)
from .hash_compressor import HashCompressor, HashLoss, VideoPositiveMask
from .object_score import (
MaskFeatures,
MaskScoringConfig,
compute_mask_features,
rank_masks,
score_mask,
select_best_mask,
)
from .pipeline import HashPipeline, create_pipeline_from_config
from .train import train
@@ -16,6 +24,12 @@ __all__ = [
"VideoPositiveMask",
"HashPipeline",
"create_pipeline_from_config",
"MaskFeatures",
"MaskScoringConfig",
"compute_mask_features",
"score_mask",
"rank_masks",
"select_best_mask",
"BinarySign",
"hamming_distance",
"hamming_similarity",