/Grid-Keypoint-Learning

Primary LanguagePython

Accurate Grid Keypoint Learning for Efficient Video Prediction

  1. Dataset processing follows https://github.com/edenton/svg create the "data" folder
  2. run train_AE_BN3_kth_1_2.py to detect keypoints
  3. run train_all_AE_BN3_kth_1_2.py to learn keypoint dynamics