Solved bug related with bounding boxes in segmentation