mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-07-13 04:25:32 +08:00
feat(compressors): add object scoring and selection for SAM masks
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user