Skip to content

Commit 5f854d0

Browse files
stephenyan1231facebook-github-bot
authored andcommitted
vis utility function visualize() support segm mask prediction
Summary: In bento N5675106 **visualize_fusion_models.py**, we'd like to support the visualization and side-by-side comparison of egobody mask predictions from different Fusion MT models. This diff contains the required changes. - For visualize() function, add a new argument **show_segm** Differential Revision: D77811204 fbshipit-source-id: 9d98dd004d9321cebdf5779e59f594af0190ee6a
1 parent ada444a commit 5f854d0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

detectron2/utils/visualizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -451,7 +451,7 @@ def draw_sem_seg(
451451
area_threshold=None,
452452
alpha=0.8,
453453
draw_text: bool = True,
454-
edge_color: tuple[float, float, float] = _OFF_WHITE,
454+
edge_color: tuple[float, float, float] | None = _OFF_WHITE,
455455
):
456456
"""
457457
Draw semantic segmentation predictions/labels.

0 commit comments

Comments
 (0)