FIX: Skip zero label in vectorization.

VHR/segmentation.py

@@ -218,13 +218,14 @@ def tiled_vectorization(input_segm, nominal_tile_size, output_template):
         tiles[i] = [np.inf, np.inf, 0, 0]
 
     for o in rp:
-        ix, iy = int(o.bbox[1] / nominal_tile_size[0]), int(o.bbox[0] / nominal_tile_size[1])
-        idx = ix * ty + iy
-        obj_to_tile[idx].append(o.label)
-        tiles[idx][0] = min(o.bbox[1], tiles[idx][0])
-        tiles[idx][1] = min(o.bbox[0], tiles[idx][1])
-        tiles[idx][2] = max(o.bbox[3], tiles[idx][2])
-        tiles[idx][3] = max(o.bbox[2], tiles[idx][3])
+        if o.label != 0:
+            ix, iy = int(o.bbox[1] / nominal_tile_size[0]), int(o.bbox[0] / nominal_tile_size[1])
+            idx = ix * ty + iy
+            obj_to_tile[idx].append(o.label)
+            tiles[idx][0] = min(o.bbox[1], tiles[idx][0])
+            tiles[idx][1] = min(o.bbox[0], tiles[idx][1])
+            tiles[idx][2] = max(o.bbox[3], tiles[idx][2])
+            tiles[idx][3] = max(o.bbox[2], tiles[idx][3])
 
     out_files = []
     for i in range(len(tiles)):