mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-07-12 20:15:31 +08:00
SikongJueluo
af0531a5eb
- switch OWLv2 loader return type to Owlv2ForObjectDetection - add detect_objects and detect_objects_batch with two-stage score filtering - add DetectionResult typed dict and conversion helper for post-processed outputs - export new detection APIs from proposal module
18 lines
390 B
Python
18 lines
390 B
Python
"""Proposal module — SAM mask generation and OWLv2 object detection."""
|
|
|
|
from .core import (
|
|
detect_objects,
|
|
detect_objects_batch,
|
|
generate_proposals,
|
|
generate_proposals_batch,
|
|
)
|
|
from .utils import extract_masked_region
|
|
|
|
__all__ = [
|
|
"detect_objects",
|
|
"detect_objects_batch",
|
|
"generate_proposals",
|
|
"generate_proposals_batch",
|
|
"extract_masked_region",
|
|
]
|