The impact of prolonged space travel isn't limited to "baby feet." Puffy faces, loss of muscle, and dizziness are other effects.
Rehab is essential for an astronaut's full recovery. They have to gradually re-acclimatize their feet to pressure and friction (walking on soft surfaces before transitioning to harder ones), restore muscle strength (specific exercises for legs and feet), and restore mobility and endurance (resistance training and stretching).