xiaoxue_admin
/
supervision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(pre_commit): 🎨 auto format pre-commit hooks

This commit is contained in:
pre-commit-ci[bot] 2025-02-18 11:52:42 +00:00
parent d0559e104d
commit e804a01e85
5 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/count_people_in_zone/inference_example.py
View File

@ -35,9 +35,7 @@ def load_zones_config(file_path: str) -> List[np.ndarray]:
def initiate_annotators(
polygons: List[np.ndarray], resolution_wh: Tuple[int, int]
) -> Tuple[
List[sv.PolygonZone], List[sv.PolygonZoneAnnotator], List[sv.BoxAnnotator]
]:
) -> Tuple[List[sv.PolygonZone], List[sv.PolygonZoneAnnotator], List[sv.BoxAnnotator]]:
line_thickness = sv.calculate_optimal_line_thickness(resolution_wh=resolution_wh)
text_scale = sv.calculate_optimal_text_scale(resolution_wh=resolution_wh)

4
examples/count_people_in_zone/ultralytics_example.py
View File

@ -33,9 +33,7 @@ def load_zones_config(file_path: str) -> List[np.ndarray]:
def initiate_annotators(
polygons: List[np.ndarray], resolution_wh: Tuple[int, int]
) -> Tuple[
List[sv.PolygonZone], List[sv.PolygonZoneAnnotator], List[sv.BoxAnnotator]
]:
) -> Tuple[List[sv.PolygonZone], List[sv.PolygonZoneAnnotator], List[sv.BoxAnnotator]]:
line_thickness = sv.calculate_optimal_line_thickness(resolution_wh=resolution_wh)
text_scale = sv.calculate_optimal_text_scale(resolution_wh=resolution_wh)

2
supervision/__init__.py
View File

@ -45,7 +45,6 @@ from supervision.detection.line_zone import (
LineZoneAnnotator,
LineZoneAnnotatorMulticlass,
)
from supervision.detection.vlm import LMM, VLM
from supervision.detection.overlap_filter import (
OverlapFilter,
box_non_max_merge,
@ -78,6 +77,7 @@ from supervision.detection.utils import (
xywh_to_xyxy,
xyxy_to_polygons,
)
from supervision.detection.vlm import LMM, VLM
from supervision.draw.color import Color, ColorPalette
from supervision.draw.utils import (
calculate_optimal_line_thickness,

1
supervision/annotators/core.py
View File

@ -36,7 +36,6 @@ from supervision.utils.image import (
overlay_image,
scale_image,
)
from supervision.utils.internal import deprecated
CV2_FONT = cv2.FONT_HERSHEY_SIMPLEX

21
supervision/detection/core.py
View File

@ -9,12 +9,6 @@ from supervision.config import (
CLASS_NAME_DATA_FIELD,
ORIENTED_BOX_COORDINATES,
)
from supervision.detection.vlm import (
LMM,
from_florence_2,
from_paligemma,
validate_vlm_parameters, VLM,
)
from supervision.detection.overlap_filter import (
box_non_max_merge,
box_non_max_suppression,
@ -38,8 +32,15 @@ from supervision.detection.utils import (
process_roboflow_result,
xywh_to_xyxy,
)
from supervision.detection.vlm import (
LMM,
VLM,
from_florence_2,
from_paligemma,
validate_vlm_parameters,
)
from supervision.geometry.core import Position
from supervision.utils.internal import get_instance_variables, deprecated
from supervision.utils.internal import deprecated, get_instance_variables
from supervision.validators import validate_detections_fields
@ -845,7 +846,7 @@ class Detections:
lmm_to_vlm = {
LMM.PALIGEMMA: VLM.PALIGEMMA,
LMM.FLORENCE_2: VLM.FLORENCE_2,
LMM.QWEN_2_5_VL: VLM.QWEN_2_5_VL
LMM.QWEN_2_5_VL: VLM.QWEN_2_5_VL,
}
if isinstance(lmm, LMM):
@ -869,7 +870,9 @@ class Detections:
return cls.from_vlm(vlm=vlm, result=result, **kwargs)
@classmethod
def from_vlm(cls, vlm: Union[VLM, str], result: Union[str, dict], **kwargs: Any) -> Detections:
def from_vlm(
cls, vlm: Union[VLM, str], result: Union[str, dict], **kwargs: Any
) -> Detections:
vlm = validate_vlm_parameters(vlm, result, kwargs)
if vlm == VLM.PALIGEMMA: