Can you add "TODO:" here so that this is not forgotten?
(or if using `scipy.ndimage` is fine, remove the comment.)