Если вы ищете современный метод обучения визуальной имитации, который обобщается на новые объекты и имеет доступный код, попробуйте его!
https://www.robot-learning.uk/dinobot
https://www.robot-learning.uk/dinobot