Add spatial segmentation