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

feat(compressors/proposal): add OWLv2 object detection pipeline and typed results

Browse Source 
- 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
This commit is contained in:
SikongJueluo
2026-04-02 19:41:54 +08:00
parent 42acb3ee1b
commit af0531a5eb
3 changed files with 157 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,9 +1,16 @@
"""Proposal module — SAM mask generation and extraction."""
"""Proposal module — SAM mask generation and OWLv2 object detection."""
from .core import generate_proposals, generate_proposals_batch
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",