Update model.py

clmpng · Oct 22, 2020 · 49e4e5c · 49e4e5c
1 parent 848bf45
commit 49e4e5c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mrcnn/model.py b/mrcnn/model.py
@@ -1278,6 +1278,7 @@ def hook(images, augmenter, parents, default):
     _idx = np.sum(mask, axis=(0, 1)) > 0
     mask = mask[:, :, _idx]
     class_ids = class_ids[_idx]
+    print(class_ids)
     # Bounding boxes. Note that some boxes might be all zeros
     # if the corresponding mask got cropped out.
     # bbox: [num_instances, (y1, x1, y2, x2)]