Skip to content

Commit

Permalink
fix(pre_commit): 🎨 auto format pre-commit hooks
Browse files Browse the repository at this point in the history
  • Loading branch information
pre-commit-ci[bot] committed Dec 16, 2024
1 parent fd4e84e commit b593c1e
Showing 1 changed file with 7 additions and 8 deletions.
15 changes: 7 additions & 8 deletions supervision/dataset/formats/yolo.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
from supervision.config import ORIENTED_BOX_COORDINATES
from supervision.dataset.utils import approximate_mask_with_polygons
from supervision.detection.core import Detections
from supervision.detection.utils import polygon_to_mask, polygon_to_xyxy
from supervision.detection.utils import polygon_to_xyxy
from supervision.utils.file import (
list_files_with_extensions,
read_txt_file,
Expand Down Expand Up @@ -44,16 +44,17 @@ def _parse_polygon(values: List[str]) -> np.ndarray:
return np.array(values, dtype=np.float32).reshape(-1, 2)


def _polygons_to_masks(polygon: list[np.ndarray], resolution_wh: Tuple[int, int]) -> np.ndarray:
def _polygons_to_masks(
polygon: list[np.ndarray], resolution_wh: Tuple[int, int]
) -> np.ndarray:
polygon_int = np.round(polygon).astype(np.int32)
mask = np.zeros((resolution_wh[1], resolution_wh[0]), dtype=np.uint8)

cv2.fillPoly(mask, [polygon_int], 1)

return mask.astype(bool)



def _with_mask(lines: List[str]) -> bool:
return any([len(line.split()) > 5 for line in lines])

Expand Down Expand Up @@ -112,9 +113,7 @@ def yolo_annotations_to_detections(
if not with_masks:
return Detections(class_id=class_id, xyxy=xyxy, data=data)

polygons = [
(polygon * np.array(resolution_wh)) for polygon in relative_polygon
]
polygons = [(polygon * np.array(resolution_wh)) for polygon in relative_polygon]
mask = _polygons_to_masks(polygon=polygons, resolution_wh=resolution_wh)
return Detections(class_id=class_id, xyxy=xyxy, data=data, mask=mask)

Expand Down Expand Up @@ -143,7 +142,7 @@ def load_yolo_annotations(
where pairs of [x, y] are box corners.
Returns:
Tuple[List[str], List[str], Dict[str, Detections]]:
A tuple containing a list of class names, a dictionary with
image names as keys and images as values, and a dictionary
Expand Down

0 comments on commit b593c1e

Please sign in to comment.