diff --git a/VHR/segmentation.py b/VHR/segmentation.py index 4cc737eb6d165704ea26771796ea83287e7f26b5..af815013435b52ce359a1f686fa96b718037566a 100644 --- a/VHR/segmentation.py +++ b/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)):