mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-07-13 04:25:32 +08:00
SikongJueluo
5f41cf5794
- 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
19 lines
471 B
Python
19 lines
471 B
Python
from .common import get_device, get_output_diretory
|
|
from .feature_extractor import (
|
|
extract_batch_features,
|
|
extract_single_image_feature,
|
|
infer_vector_dim,
|
|
)
|
|
from .image import crop_image_by_bbox, extract_masked_region, numpy_to_pil
|
|
|
|
__all__ = [
|
|
"get_device",
|
|
"get_output_diretory",
|
|
"infer_vector_dim",
|
|
"extract_single_image_feature",
|
|
"extract_batch_features",
|
|
"numpy_to_pil",
|
|
"extract_masked_region",
|
|
"crop_image_by_bbox",
|
|
]
|