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

Update utils.py

parent c53d5e6e
......@@ -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.permute(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