Commit 1376176c authored by Lamping, Christian's avatar Lamping, Christian
Browse files

Update utils.py

parent 2c5879b6
......@@ -617,6 +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)
# 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