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

Update utils.py

parent 1376176c
......@@ -315,7 +315,7 @@ class Dataset(object):
min_scale=None,
max_dim=384)
mask = resize_mask(mask, scale, padding, crop)
mask = mask.transpose(2,0,1)
mask = mask.transpose((2,0,1))
return mask, class_ids, scores
def resize(image, output_shape, order=1, mode='constant', cval=0, clip=True,
......@@ -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