vqdang#90 UPD: add more shape check

samjmolyneux · Jan 19, 2021 · d801137 · d801137
1 parent 0845678
commit d801137
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/models/hovernet/post_proc.py b/models/hovernet/post_proc.py
@@ -139,6 +139,8 @@ def process(pred_map, nr_types=None, return_centroids=False):
             # as the contours obtained via approximation => too small or sthg
             if inst_contour.shape[0] < 3:
                 continue
+            if len(inst_contour.shape) != 2:
+                continue # ! check for trickery shape
             inst_centroid = [
                 (inst_moment["m10"] / inst_moment["m00"]),
                 (inst_moment["m01"] / inst_moment["m00"]),