Using attention for video segmentation