Skip to content

Commit 3503a63

Browse files
update docstring for 'fill_labels'
1 parent 2c4e798 commit 3503a63

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torchvision/utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -322,7 +322,8 @@ def draw_bounding_boxes(
322322
`colors` argument for details. Defaults to the same colors used for the boxes, or to black if ``fill_labels`` is True.
323323
label_background_colors (color or list of colors, optional): Colors for the label text box fill. Defaults to the
324324
same colors used for the boxes. Ignored when ``fill_labels`` is False.
325-
fill_labels (bool): If `True` fills the label background with specified box color (from the ``colors`` parameter). Default: False.
325+
fill_labels (bool): If `True` fills the label background with specified color (from the ``label_background_colors`` parameter,
326+
or from the ``colors`` parameter if not specified). Default: False.
326327
327328
Returns:
328329
img (Tensor[C, H, W]): Image Tensor of dtype uint8 with bounding boxes plotted.

0 commit comments

Comments
 (0)