Skip to content

Commit 1c83989

Browse files
committed
docs and examples
1 parent 3e7e4ca commit 1c83989

File tree

2 files changed

+40
-0
lines changed

2 files changed

+40
-0
lines changed

docs/detection/utils/converters.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,3 +58,9 @@ status: new
5858
</div>
5959

6060
:::supervision.detection.utils.converters.polygon_to_xyxy
61+
62+
<div class="md-typeset">
63+
<h2><a href="#supervision.detection.utils.converters.xyxy_to_mask">xyxy_to_mask</a></h2>
64+
</div>
65+
66+
:::supervision.detection.utils.converters.xyxy_to_mask

supervision/detection/utils/converters.py

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -242,6 +242,40 @@ def xyxy_to_mask(boxes: np.ndarray, resolution_wh: tuple[int, int]) -> np.ndarra
242242
Returns:
243243
np.ndarray: A 3D `np.ndarray` of shape `(N, height, width)`
244244
containing 2D bool masks for each bounding box
245+
246+
Examples:
247+
```python
248+
import numpy as np
249+
import supervision as sv
250+
251+
boxes = np.array([[0, 0, 2, 2]])
252+
253+
sv.xyxy_to_mask(boxes, (5, 5))
254+
# array([
255+
# [[ True, True, True, False, False],
256+
# [ True, True, True, False, False],
257+
# [ True, True, True, False, False],
258+
# [False, False, False, False, False],
259+
# [False, False, False, False, False]]
260+
# ])
261+
262+
boxes = np.array([[0, 0, 1, 1], [3, 3, 4, 4]])
263+
264+
sv.xyxy_to_mask(boxes, (5, 5))
265+
# array([
266+
# [[ True, True, False, False, False],
267+
# [ True, True, False, False, False],
268+
# [False, False, False, False, False],
269+
# [False, False, False, False, False],
270+
# [False, False, False, False, False]],
271+
#
272+
# [[False, False, False, False, False],
273+
# [False, False, False, False, False],
274+
# [False, False, False, False, False],
275+
# [False, False, False, True, True],
276+
# [False, False, False, True, True]]
277+
# ])
278+
```
245279
"""
246280
width, height = resolution_wh
247281
n = boxes.shape[0]

0 commit comments

Comments
 (0)