Just expend the following line :https://gitlab.nibio.no/maciekwielgosz/instance_segmentation_classic/-/blob/master/nibio_preprocessing/add_ground_to_inst_seg.py#L50