`new_segment_map = np.zeros_like(segment_map)`?