ENH: Add an option to select 2D or 3D image registration.