Update utils.py

utils.py

@@ -586,6 +586,9 @@ def display_top_masks(image, mask, class_ids, class_names, limit=4):
     """Display the given image and the top few class masks."""
     to_display = []
     titles = []
+    image = np.array(image)
+    image = image.transpose(1, 2, 0)
+    mask = mask.transpose(1, 2, 0)
     to_display.append(image)
     titles.append("H x W={}x{}".format(image.shape[0], image.shape[1]))
     # Pick top prominent classes in this image