Skip to content

Commit c55d381

Browse files
add cpp wrapper for QVE
Summary: Pull Request resolved: #3525 Add c++ wrapper for instance segmentation Reviewed By: zhanghang1989 Differential Revision: D31049697 fbshipit-source-id: 37db567468d1a72f73d3f8834684895e18666f0b
1 parent f64c764 commit c55d381

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
@@ -86,7 +86,7 @@ def __init__(self, mask_or_polygons, height, width):
8686

8787
if isinstance(m, np.ndarray): # assumed to be a binary mask
8888
assert m.shape[1] != 2, m.shape
89-
assert m.shape == (height, width), m.shape
89+
assert m.shape == (height, width), f"mask shape: {m.shape}, target dims: {height}, {width}"
9090
self._mask = m.astype("uint8")
9191
return
9292

0 commit comments

Comments
 (0)