Skip to content

Commit 046073b

Browse files
Added kwargs and label_border_color to end2end_detect() (airctic#891)
* added kwargs and label_border_color * no need to pass label_border_color
1 parent 773a702 commit 046073b

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

icevision/models/inference.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,10 @@ def _end2end_detect(
2727
display_score: bool = True,
2828
font_path: Optional[os.PathLike] = DEFAULT_FONT_PATH,
2929
font_size: Union[int, float] = 12,
30-
label_color: Union[np.array, list, tuple, str] = ("#C4C4C4"), # Mild Gray
30+
label_color: Union[np.array, list, tuple, str] = ("#FF59D6"), # Pink
3131
return_as_pil_img=True,
3232
return_img=True,
33+
**kwargs,
3334
):
3435
"""
3536
Run Object Detection inference (only `bboxes`) on a single image.
@@ -77,6 +78,7 @@ def _end2end_detect(
7778
font_size=font_size,
7879
label_color=label_color,
7980
return_as_pil_img=return_as_pil_img,
81+
**kwargs,
8082
)
8183
record.set_img(pred_img)
8284
else:

0 commit comments

Comments
 (0)