Update spectral_seg.py