ATK: Automatic Task-driven Keypoint Selection for Robust Policy Learning
Learning visuamotor policy through imitation learning often suffers from perceptual challenges, where visual differences between training and evaluation environments degrade policy performance. Policies relying on state estimations like 6D pose, require task-specific tracking and are difficult to sc…