Add semantic segmentation (linear) code + demo notebook