├── LICENSE ├── README.md ├── media ├── 1st_order_2D.png ├── 1st_order_2D_diffeo.png ├── 1st_order_3D.png ├── 1st_order_ND.png ├── 2nd_order_2D.png ├── robot_demo.gif └── s_shape_animation.gif ├── poetry.lock ├── pyproject.toml └── src ├── agent ├── contrastive_imitation.py ├── dynamical_system.py ├── neural_network.py └── utils │ ├── dynamical_system_operations.py │ └── ranking_losses.py ├── data_preprocessing ├── data_loader.py └── data_preprocessor.py ├── datasets ├── LAIR │ ├── G_angle │ │ ├── G_angle_0.npy │ │ ├── G_angle_1.npy │ │ ├── G_angle_10.npy │ │ ├── G_angle_11.npy │ │ ├── G_angle_12.npy │ │ ├── G_angle_13.npy │ │ ├── G_angle_2.npy │ │ ├── G_angle_3.npy │ │ ├── G_angle_4.npy │ │ ├── G_angle_5.npy │ │ ├── G_angle_6.npy │ │ ├── G_angle_7.npy │ │ ├── G_angle_8.npy │ │ └── G_angle_9.npy │ ├── Lag │ │ ├── Lag_0.npy │ │ ├── Lag_1.npy │ │ ├── Lag_2.npy │ │ ├── Lag_3.npy │ │ ├── Lag_4.npy │ │ ├── Lag_5.npy │ │ └── Lag_6.npy │ ├── capricorn │ │ ├── capricorn_0.npy │ │ ├── capricorn_1.npy │ │ ├── capricorn_2.npy │ │ ├── capricorn_3.npy │ │ ├── capricorn_4.npy │ │ ├── capricorn_5.npy │ │ └── capricorn_6.npy │ ├── double_lag │ │ ├── double_lag_0.npy │ │ ├── double_lag_1.npy │ │ ├── double_lag_2.npy │ │ ├── double_lag_3.npy │ │ ├── double_lag_4.npy │ │ ├── double_lag_5.npy │ │ └── double_lag_6.npy │ ├── double_loop │ │ ├── double_loop_0.npy │ │ ├── double_loop_1.npy │ │ ├── double_loop_2.npy │ │ ├── double_loop_3.npy │ │ ├── double_loop_4.npy │ │ ├── double_loop_5.npy │ │ └── double_loop_6.npy │ ├── e │ │ ├── e_0.npy │ │ ├── e_1.npy │ │ ├── e_2.npy │ │ ├── e_3.npy │ │ ├── e_4.npy │ │ ├── e_5.npy │ │ └── e_6.npy │ ├── mountain │ │ ├── mountain_0.npy │ │ ├── mountain_1.npy │ │ ├── mountain_10.npy │ │ ├── mountain_11.npy │ │ ├── mountain_12.npy │ │ ├── mountain_13.npy │ │ ├── mountain_2.npy │ │ ├── mountain_3.npy │ │ ├── mountain_4.npy │ │ ├── mountain_5.npy │ │ ├── mountain_6.npy │ │ ├── mountain_7.npy │ │ ├── mountain_8.npy │ │ └── mountain_9.npy │ ├── phi │ │ ├── phi_0.npy │ │ ├── phi_1.npy │ │ ├── phi_2.npy │ │ ├── phi_3.npy │ │ ├── phi_4.npy │ │ ├── phi_5.npy │ │ └── phi_6.npy │ ├── triple_loop │ │ ├── triple_loop_0.npy │ │ ├── triple_loop_1.npy │ │ ├── triple_loop_2.npy │ │ ├── triple_loop_3.npy │ │ ├── triple_loop_4.npy │ │ ├── triple_loop_5.npy │ │ └── triple_loop_6.npy │ ├── two │ │ ├── two_0.npy │ │ ├── two_1.npy │ │ ├── two_2.npy │ │ ├── two_3.npy │ │ ├── two_4.npy │ │ └── two_5.npy │ └── two_roads │ │ ├── two_roads_0.npy │ │ ├── two_roads_1.npy │ │ ├── two_roads_10.npy │ │ ├── two_roads_11.npy │ │ ├── two_roads_12.npy │ │ ├── two_roads_13.npy │ │ ├── two_roads_2.npy │ │ ├── two_roads_3.npy │ │ ├── two_roads_4.npy │ │ ├── two_roads_5.npy │ │ ├── two_roads_6.npy │ │ ├── two_roads_7.npy │ │ ├── two_roads_8.npy │ │ └── two_roads_9.npy ├── LASA │ ├── Angle.mat │ ├── BendedLine.mat │ ├── CShape.mat │ ├── DoubleBendedLine.mat │ ├── GShape.mat │ ├── JShape.mat │ ├── JShape_2.mat │ ├── Khamesh.mat │ ├── LShape.mat │ ├── Leaf_1.mat │ ├── Leaf_2.mat │ ├── Line.mat │ ├── Multi_Models_1.mat │ ├── Multi_Models_2.mat │ ├── Multi_Models_3.mat │ ├── Multi_Models_4.mat │ ├── NShape.mat │ ├── PShape.mat │ ├── RShape.mat │ ├── Saeghe.mat │ ├── Sharpc.mat │ ├── Sine.mat │ ├── Snake.mat │ ├── Spoon.mat │ ├── Sshape.mat │ ├── Trapezoid.mat │ ├── WShape.mat │ ├── Worm.mat │ ├── Zshape.mat │ └── heee.mat ├── dataset_keys.py ├── interpolation │ ├── interpolation_1 │ │ ├── interpolation_0.npy │ │ ├── interpolation_1.npy │ │ ├── interpolation_2.npy │ │ ├── interpolation_3.npy │ │ └── interpolation_4.npy │ ├── interpolation_2 │ │ ├── interpolation_5.npy │ │ ├── interpolation_6.npy │ │ ├── interpolation_7.npy │ │ ├── interpolation_8.npy │ │ └── interpolation_9.npy │ └── interpolation_3 │ │ ├── interpolation_10.npy │ │ ├── interpolation_11.npy │ │ ├── interpolation_12.npy │ │ ├── interpolation_13.npy │ │ └── interpolation_14.npy ├── joint_space │ ├── cleaning_1 │ │ └── joints_state_0.pk │ └── cleaning_2 │ │ └── joints_state_0.pk └── optitrack │ └── hammer │ ├── positions_0.npy │ ├── positions_1.npy │ ├── positions_2.npy │ ├── positions_3.npy │ ├── positions_4.npy │ ├── positions_5.npy │ ├── positions_6.npy │ ├── positions_7.npy │ ├── positions_8.npy │ └── positions_9.npy ├── evaluation ├── evaluate.py ├── evaluate_2d_o1.py ├── evaluate_2d_o2.py ├── evaluate_3d.py ├── evaluate_nd.py ├── evaluator_init.py └── utils │ ├── saving.py │ └── similarity_measures.py ├── hyperparam_optimization ├── create_optuna_study.py ├── optuna_functions.py └── visualize_hyperparameter_optimization.py ├── initializer.py ├── params ├── 1st_order_2D.py ├── 1st_order_2D_fixed_gains.py ├── 1st_order_2D_triplet.py ├── 1st_order_3D.py ├── 1st_order_joint_space.py ├── 1st_order_multi.py ├── 2nd_order_2D.py └── behavioral_cloning.py ├── run_optuna.py ├── simulate_ds.py ├── tools ├── animation.py ├── draw_mouse.py └── plot_3D.py └── train.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/README.md -------------------------------------------------------------------------------- /media/1st_order_2D.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/media/1st_order_2D.png -------------------------------------------------------------------------------- /media/1st_order_2D_diffeo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/media/1st_order_2D_diffeo.png -------------------------------------------------------------------------------- /media/1st_order_3D.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/media/1st_order_3D.png -------------------------------------------------------------------------------- /media/1st_order_ND.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/media/1st_order_ND.png -------------------------------------------------------------------------------- /media/2nd_order_2D.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/media/2nd_order_2D.png -------------------------------------------------------------------------------- /media/robot_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/media/robot_demo.gif -------------------------------------------------------------------------------- /media/s_shape_animation.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/media/s_shape_animation.gif -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/pyproject.toml -------------------------------------------------------------------------------- /src/agent/contrastive_imitation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/agent/contrastive_imitation.py -------------------------------------------------------------------------------- /src/agent/dynamical_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/agent/dynamical_system.py -------------------------------------------------------------------------------- /src/agent/neural_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/agent/neural_network.py -------------------------------------------------------------------------------- /src/agent/utils/dynamical_system_operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/agent/utils/dynamical_system_operations.py -------------------------------------------------------------------------------- /src/agent/utils/ranking_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/agent/utils/ranking_losses.py -------------------------------------------------------------------------------- /src/data_preprocessing/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/data_preprocessing/data_loader.py -------------------------------------------------------------------------------- /src/data_preprocessing/data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/data_preprocessing/data_preprocessor.py -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_0.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_1.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_10.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_10.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_11.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_11.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_12.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_12.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_13.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_13.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_2.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_3.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_4.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_5.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_6.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_7.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_7.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_8.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_8.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/G_angle/G_angle_9.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/G_angle/G_angle_9.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/Lag/Lag_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/Lag/Lag_0.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/Lag/Lag_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/Lag/Lag_1.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/Lag/Lag_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/Lag/Lag_2.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/Lag/Lag_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/Lag/Lag_3.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/Lag/Lag_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/Lag/Lag_4.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/Lag/Lag_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/Lag/Lag_5.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/Lag/Lag_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/Lag/Lag_6.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/capricorn/capricorn_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/capricorn/capricorn_0.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/capricorn/capricorn_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/capricorn/capricorn_1.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/capricorn/capricorn_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/capricorn/capricorn_2.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/capricorn/capricorn_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/capricorn/capricorn_3.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/capricorn/capricorn_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/capricorn/capricorn_4.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/capricorn/capricorn_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/capricorn/capricorn_5.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/capricorn/capricorn_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/capricorn/capricorn_6.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_lag/double_lag_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_lag/double_lag_0.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_lag/double_lag_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_lag/double_lag_1.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_lag/double_lag_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_lag/double_lag_2.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_lag/double_lag_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_lag/double_lag_3.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_lag/double_lag_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_lag/double_lag_4.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_lag/double_lag_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_lag/double_lag_5.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_lag/double_lag_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_lag/double_lag_6.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_loop/double_loop_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_loop/double_loop_0.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_loop/double_loop_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_loop/double_loop_1.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_loop/double_loop_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_loop/double_loop_2.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_loop/double_loop_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_loop/double_loop_3.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_loop/double_loop_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_loop/double_loop_4.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_loop/double_loop_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_loop/double_loop_5.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/double_loop/double_loop_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/double_loop/double_loop_6.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/e/e_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/e/e_0.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/e/e_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/e/e_1.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/e/e_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/e/e_2.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/e/e_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/e/e_3.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/e/e_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/e/e_4.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/e/e_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/e/e_5.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/e/e_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/e/e_6.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_0.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_1.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_10.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_10.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_11.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_11.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_12.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_12.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_13.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_13.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_2.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_3.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_4.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_5.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_6.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_7.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_7.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_8.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_8.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/mountain/mountain_9.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/mountain/mountain_9.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/phi/phi_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/phi/phi_0.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/phi/phi_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/phi/phi_1.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/phi/phi_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/phi/phi_2.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/phi/phi_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/phi/phi_3.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/phi/phi_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/phi/phi_4.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/phi/phi_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/phi/phi_5.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/phi/phi_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/phi/phi_6.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/triple_loop/triple_loop_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/triple_loop/triple_loop_0.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/triple_loop/triple_loop_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/triple_loop/triple_loop_1.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/triple_loop/triple_loop_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/triple_loop/triple_loop_2.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/triple_loop/triple_loop_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/triple_loop/triple_loop_3.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/triple_loop/triple_loop_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/triple_loop/triple_loop_4.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/triple_loop/triple_loop_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/triple_loop/triple_loop_5.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/triple_loop/triple_loop_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/triple_loop/triple_loop_6.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two/two_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two/two_0.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two/two_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two/two_1.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two/two_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two/two_2.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two/two_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two/two_3.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two/two_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two/two_4.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two/two_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two/two_5.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_0.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_1.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_10.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_10.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_11.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_11.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_12.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_12.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_13.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_13.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_2.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_3.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_4.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_5.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_6.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_7.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_7.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_8.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_8.npy -------------------------------------------------------------------------------- /src/datasets/LAIR/two_roads/two_roads_9.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LAIR/two_roads/two_roads_9.npy -------------------------------------------------------------------------------- /src/datasets/LASA/Angle.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Angle.mat -------------------------------------------------------------------------------- /src/datasets/LASA/BendedLine.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/BendedLine.mat -------------------------------------------------------------------------------- /src/datasets/LASA/CShape.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/CShape.mat -------------------------------------------------------------------------------- /src/datasets/LASA/DoubleBendedLine.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/DoubleBendedLine.mat -------------------------------------------------------------------------------- /src/datasets/LASA/GShape.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/GShape.mat -------------------------------------------------------------------------------- /src/datasets/LASA/JShape.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/JShape.mat -------------------------------------------------------------------------------- /src/datasets/LASA/JShape_2.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/JShape_2.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Khamesh.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Khamesh.mat -------------------------------------------------------------------------------- /src/datasets/LASA/LShape.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/LShape.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Leaf_1.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Leaf_1.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Leaf_2.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Leaf_2.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Line.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Line.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Multi_Models_1.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Multi_Models_1.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Multi_Models_2.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Multi_Models_2.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Multi_Models_3.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Multi_Models_3.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Multi_Models_4.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Multi_Models_4.mat -------------------------------------------------------------------------------- /src/datasets/LASA/NShape.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/NShape.mat -------------------------------------------------------------------------------- /src/datasets/LASA/PShape.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/PShape.mat -------------------------------------------------------------------------------- /src/datasets/LASA/RShape.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/RShape.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Saeghe.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Saeghe.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Sharpc.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Sharpc.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Sine.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Sine.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Snake.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Snake.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Spoon.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Spoon.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Sshape.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Sshape.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Trapezoid.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Trapezoid.mat -------------------------------------------------------------------------------- /src/datasets/LASA/WShape.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/WShape.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Worm.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Worm.mat -------------------------------------------------------------------------------- /src/datasets/LASA/Zshape.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/Zshape.mat -------------------------------------------------------------------------------- /src/datasets/LASA/heee.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/LASA/heee.mat -------------------------------------------------------------------------------- /src/datasets/dataset_keys.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/dataset_keys.py -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_1/interpolation_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_1/interpolation_0.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_1/interpolation_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_1/interpolation_1.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_1/interpolation_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_1/interpolation_2.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_1/interpolation_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_1/interpolation_3.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_1/interpolation_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_1/interpolation_4.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_2/interpolation_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_2/interpolation_5.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_2/interpolation_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_2/interpolation_6.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_2/interpolation_7.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_2/interpolation_7.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_2/interpolation_8.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_2/interpolation_8.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_2/interpolation_9.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_2/interpolation_9.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_3/interpolation_10.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_3/interpolation_10.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_3/interpolation_11.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_3/interpolation_11.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_3/interpolation_12.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_3/interpolation_12.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_3/interpolation_13.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_3/interpolation_13.npy -------------------------------------------------------------------------------- /src/datasets/interpolation/interpolation_3/interpolation_14.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/interpolation/interpolation_3/interpolation_14.npy -------------------------------------------------------------------------------- /src/datasets/joint_space/cleaning_1/joints_state_0.pk: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/joint_space/cleaning_1/joints_state_0.pk -------------------------------------------------------------------------------- /src/datasets/joint_space/cleaning_2/joints_state_0.pk: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/joint_space/cleaning_2/joints_state_0.pk -------------------------------------------------------------------------------- /src/datasets/optitrack/hammer/positions_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/optitrack/hammer/positions_0.npy -------------------------------------------------------------------------------- /src/datasets/optitrack/hammer/positions_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/optitrack/hammer/positions_1.npy -------------------------------------------------------------------------------- /src/datasets/optitrack/hammer/positions_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/optitrack/hammer/positions_2.npy -------------------------------------------------------------------------------- /src/datasets/optitrack/hammer/positions_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/optitrack/hammer/positions_3.npy -------------------------------------------------------------------------------- /src/datasets/optitrack/hammer/positions_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/optitrack/hammer/positions_4.npy -------------------------------------------------------------------------------- /src/datasets/optitrack/hammer/positions_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/optitrack/hammer/positions_5.npy -------------------------------------------------------------------------------- /src/datasets/optitrack/hammer/positions_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/optitrack/hammer/positions_6.npy -------------------------------------------------------------------------------- /src/datasets/optitrack/hammer/positions_7.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/optitrack/hammer/positions_7.npy -------------------------------------------------------------------------------- /src/datasets/optitrack/hammer/positions_8.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/optitrack/hammer/positions_8.npy -------------------------------------------------------------------------------- /src/datasets/optitrack/hammer/positions_9.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/datasets/optitrack/hammer/positions_9.npy -------------------------------------------------------------------------------- /src/evaluation/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/evaluation/evaluate.py -------------------------------------------------------------------------------- /src/evaluation/evaluate_2d_o1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/evaluation/evaluate_2d_o1.py -------------------------------------------------------------------------------- /src/evaluation/evaluate_2d_o2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/evaluation/evaluate_2d_o2.py -------------------------------------------------------------------------------- /src/evaluation/evaluate_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/evaluation/evaluate_3d.py -------------------------------------------------------------------------------- /src/evaluation/evaluate_nd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/evaluation/evaluate_nd.py -------------------------------------------------------------------------------- /src/evaluation/evaluator_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/evaluation/evaluator_init.py -------------------------------------------------------------------------------- /src/evaluation/utils/saving.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/evaluation/utils/saving.py -------------------------------------------------------------------------------- /src/evaluation/utils/similarity_measures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/evaluation/utils/similarity_measures.py -------------------------------------------------------------------------------- /src/hyperparam_optimization/create_optuna_study.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/hyperparam_optimization/create_optuna_study.py -------------------------------------------------------------------------------- /src/hyperparam_optimization/optuna_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/hyperparam_optimization/optuna_functions.py -------------------------------------------------------------------------------- /src/hyperparam_optimization/visualize_hyperparameter_optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/hyperparam_optimization/visualize_hyperparameter_optimization.py -------------------------------------------------------------------------------- /src/initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/initializer.py -------------------------------------------------------------------------------- /src/params/1st_order_2D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/params/1st_order_2D.py -------------------------------------------------------------------------------- /src/params/1st_order_2D_fixed_gains.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/params/1st_order_2D_fixed_gains.py -------------------------------------------------------------------------------- /src/params/1st_order_2D_triplet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/params/1st_order_2D_triplet.py -------------------------------------------------------------------------------- /src/params/1st_order_3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/params/1st_order_3D.py -------------------------------------------------------------------------------- /src/params/1st_order_joint_space.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/params/1st_order_joint_space.py -------------------------------------------------------------------------------- /src/params/1st_order_multi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/params/1st_order_multi.py -------------------------------------------------------------------------------- /src/params/2nd_order_2D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/params/2nd_order_2D.py -------------------------------------------------------------------------------- /src/params/behavioral_cloning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/params/behavioral_cloning.py -------------------------------------------------------------------------------- /src/run_optuna.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/run_optuna.py -------------------------------------------------------------------------------- /src/simulate_ds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/simulate_ds.py -------------------------------------------------------------------------------- /src/tools/animation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/tools/animation.py -------------------------------------------------------------------------------- /src/tools/draw_mouse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/tools/draw_mouse.py -------------------------------------------------------------------------------- /src/tools/plot_3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/tools/plot_3D.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rperezdattari/Stable-Motion-Primitives-via-Imitation-and-Contrastive-Learning/HEAD/src/train.py --------------------------------------------------------------------------------