mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-07-13 04:25:32 +08:00
feat(compressors): add OWLv2 bbox crop to HashPipeline and refactor image utilities
- Add crop_batch method to HashPipeline for cropping images using OWLv2 detection boxes - Integrate crop_batch into pipeline forward pass (extract_hash and extract_features) - Move extract_masked_region from compressors/proposal/utils.py to utils/image.py - Add crop_image_by_bbox utility function in utils/image.py - Update type annotations to use dict[str, Any] instead of bare dict - Update .justfile to add memory server command - Update marimo dependency to >=0.22.0 - Update nvidia CUDA package markers for platform compatibility
This commit is contained in:
@@ -4,7 +4,7 @@ from .feature_extractor import (
|
||||
extract_single_image_feature,
|
||||
infer_vector_dim,
|
||||
)
|
||||
from .image import numpy_to_pil
|
||||
from .image import crop_image_by_bbox, extract_masked_region, numpy_to_pil
|
||||
|
||||
__all__ = [
|
||||
"get_device",
|
||||
@@ -13,4 +13,6 @@ __all__ = [
|
||||
"extract_single_image_feature",
|
||||
"extract_batch_features",
|
||||
"numpy_to_pil",
|
||||
"extract_masked_region",
|
||||
"crop_image_by_bbox",
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user