Commit c53d5e6e authored by Lamping, Christian's avatar Lamping, Christian
Browse files

Update utils.py

parent b4847dfe
......@@ -617,7 +617,7 @@ def display_instances(image, boxes, masks, class_ids, class_names,
"""
image = np.array(image)
boxes = np.array(boxes)
masks = masks.transpose((1,2,0))
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