Generalizing Object-Centric Task-Axes Controllers using Keypoints