mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-07-13 04:25:32 +08:00
SikongJueluo
4e16e38f32
- Replace bbox-prompted segmentation with OWLv2 text-guided object detection - Refactor HashPipeline from nn.Module to plain class with modular stage methods - Add detect_batch, segment_batch, filter_batch for explicit pipeline stages - Rename forward to forward_batch with text_labels API instead of bboxes - Add mask_scoring_config, score_threshold, postprocess_threshold configuration - Update model_loader to expose Dinov2Model type annotation
39 lines
818 B
Python
39 lines
818 B
Python
from .common import (
|
|
BinarySign,
|
|
bits_to_hash,
|
|
hamming_distance,
|
|
hamming_similarity,
|
|
hash_to_bits,
|
|
)
|
|
from .hash_compressor import HashCompressor, HashLoss, VideoPositiveMask
|
|
from .filter 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
|
|
|
|
__all__ = [
|
|
"train",
|
|
"HashCompressor",
|
|
"HashLoss",
|
|
"VideoPositiveMask",
|
|
"HashPipeline",
|
|
"create_pipeline_from_config",
|
|
"MaskFeatures",
|
|
"MaskScoringConfig",
|
|
"compute_mask_features",
|
|
"score_mask",
|
|
"rank_masks",
|
|
"select_best_mask",
|
|
"BinarySign",
|
|
"hamming_distance",
|
|
"hamming_similarity",
|
|
"bits_to_hash",
|
|
"hash_to_bits",
|
|
]
|