Commit 23e7823e authored by Lamping, Christian's avatar Lamping, Christian
Browse files

Update utils.py

parent 9713bdbb
......@@ -696,6 +696,7 @@ def extract_bboxes(mask):
mask: [height, width, num_instances]. Mask pixels are either 1 or 0.
Returns: bbox array [num_instances, (y1, x1, y2, x2)].
"""
mask = mask.transpose((1,2,0))
boxes = np.zeros([mask.shape[-1], 4], dtype=np.int32)
for i in range(mask.shape[-1]):
m = mask[:, :, i]
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment