Commit 8f0ba2c1 authored by Lamping, Christian's avatar Lamping, Christian
Browse files

Update utils.py

parent 19805b84
......@@ -622,7 +622,8 @@ def display_instances(image, boxes, masks, class_ids, class_names,
"""
image = np.array(image)
boxes = np.array(boxes)
masks = masks.permute(1, 2, 0)
masks = np.array(masks)
masks = masks.transpose(1, 2, 0)
# Number of instances
N = boxes.shape[0]
if not N:
......
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