ENH: allow applying segmentation or not ....