Update 0_split_rgb.py