Models uploaded

trained_models/hmp/amass-mano/autoencoder/checkpoints/cvpr_release.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8b9cf7d9ea512af8d82779a835f25a0b1f8d3e11d73b37d732ff97936b8bc12a
+size 111366357

trained_models/hmp/amass-mano/autoencoder/config.yaml

@@ -0,0 +1,113 @@
+dataset_main_path: ./datasets
+dataset_annotation_path: ./datasets/annotations
+dataset_precomputed_path: ./datasets/processed
+resume_training: false
+device: cuda
+load: false
+load_path: ''
+output_log_path: ./trained_models/hmp/amass-mano/autoencoder
+batch_size: 64
+batch_size_eval: 512
+if_run_validation: False
+eval_frequency: 25
+num_epochs: 300
+num_iteration_eval: 10
+num_workers: 4
+seed: 52345
+use_lr_scheduler: true
+lr_scheduler_kwargs:
+  lr_scheduler_type: ExponentialLRSchedulerWarmup
+  warmup_duration: 10
+  update_every: 1
+  min_lr: 0.0001
+  gamma_decay: 0.98
+loss_pose_type: l1
+lr: 0.005
+latent_size: 96
+output_size: 3
+z_activation: tanh
+num_iter_perepoch: 580
+obs_length: 30
+prediction_horizon_train: 120
+prediction_horizon_eval: 120
+prediction_horizon: 120
+pred_length: 120
+prediction_horizon_train_min: 10
+prediction_horizon_train_min_from_epoch: 200
+curriculum_it: 10
+random_prediction_horizon: true
+if_consider_hip: false
+history_sec: 0.5
+prediction_horizon_sec: 2
+motion_repr_type: SkeletonRescalePose
+pose_box_size: 1.2
+seq_centering: 0
+task_name: hmp
+num_joints: 52
+fps: 60
+multimodal_threshold: 0.4
+dataset_type: AMASSDataset
+dataset_name: amass-mano
+precomputed_folder: ./datasets/processed/AMASS-MANO/hmp
+annotations_folder: ./datasets/annotations/AMASS-MANO/hmp
+dtype: float32
+data_loader_train:
+  stride: 60
+  augmentation: 30
+  shuffle: true
+  datasets:
+  - ACCAD
+  - BMLhandball
+  - BMLmovi
+  - BMLrub
+  - EKUT
+  - CMU
+  - EyesJapanDataset
+  - KIT
+  - PosePrior
+  - TCDHands
+  - TotalCapture
+  da_mirroring: 0.5
+  da_rotations: 1.0
+  drop_last: true
+  if_load_mmgt: false
+data_loader_train_eval:
+  stride: 30
+  augmentation: 0
+  shuffle: false
+  datasets:
+  - ACCAD
+  - BMLhandball
+  - BMLmovi
+  - BMLrub
+  - EKUT
+  - CMU
+  - EyesJapanDataset
+  - KIT
+  - PosePrior
+  - TCDHands
+  - TotalCapture
+  da_mirroring: 0.0
+  da_rotations: 0.0
+  drop_last: false
+  if_load_mmgt: false
+data_loader_valid:
+  stride: 30
+  augmentation: 0
+  shuffle: false
+  datasets:
+  - HumanEva
+  - HDM05
+  - SFU
+  - MoSh
+  file_idces: all
+  drop_last: false
+  if_load_mmgt: false
+
+autoenc_arch:
+  enc_num_layers: 2
+  encoder_hidden_size: 96
+  decoder_hidden_size: 96
+  arch: AutoEncoder
+  recurrent_arch_enc: StaticGraphGRU
+  recurrent_arch_decoder: StaticGraphGRU

trained_models/hmp/amass-mano/diffusion/checkpoints/cvpr_release.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2a894bd0f79fa0accf46a6dd0e2686c47d2bc6796c03f0277b187a8cef12855e
+size 394796990

trained_models/hmp/amass-mano/diffusion/config.yaml

@@ -0,0 +1,56 @@
+if_resume_training: false
+device: cuda
+load: false
+load_path: ''
+dataset_main_path: ./datasets
+dataset_annotation_path: ./datasets/annotations
+dataset_precomputed_path: ./datasets/processed
+output_log_path: ./trained_models/hmp/amass-mano/diffusion
+pretrained_autoencoder_path: ./trained_models/hmp/amass-mano/autoencoder/checkpoints/cvpr_release.pt
+lr: 0.001
+diffusion_objective: pred_x0
+weight_decay: 0.0
+if_use_ema: true
+step_start_ema: 100
+ema_power: 0.6666666666666666
+ema_update_every: 10
+ema_min_value: 0.0
+use_lr_scheduler: true
+lr_scheduler_kwargs:
+  lr_scheduler_type: ExponentialLRSchedulerWarmup
+  warmup_duration: 75
+  update_every: 2
+  min_lr: 0.0001
+  gamma_decay: 0.98
+diffusion_conditioning: true
+num_epochs: 150
+num_workers: 4
+batch_size: 64
+batch_size_eval: 256
+eval_frequency: 25
+train_pick_best_sample_among_k: 50
+similarity_space: input_space
+diffusion_activation: identity
+num_prob_samples: 50
+diffusion_timesteps: 10
+diffusion_type: NonisotropicGaussianDiffusion
+diffusion_loss_type: snr
+loss_reduction_type: l1
+if_run_as_isotropic: false
+if_sigma_n_scale: true
+diffusion_covariance_type: skeleton-diffusion
+gamma_scheduler: cosine
+beta_schedule: cosine
+sigma_n_scale: spectral
+num_iter_perepoch: null
+seed: 63485
+diffusion_arch:
+  arch: Denoiser
+  use_attention: true
+  self_condition: false
+  norm_type: none
+  depth: 4
+  attn_dim_head: 32
+  attn_heads: 8
+  learn_influence: true
+covariance_matrix_type: adjacency

trained_models/hmp/amass/autoencoder/checkpoints/cvpr_release.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7f206a4108ce6ba1beec3f3e70f6b6092fb02a57c48f6c4b462a5d366d9518c5
+size 33673173

trained_models/hmp/amass/autoencoder/config.yaml

@@ -0,0 +1,114 @@
+dataset_main_path: ./datasets
+dataset_annotation_path: ./datasets/annotations
+dataset_precomputed_path: ./datasets/processed
+resume_training: false
+device: cuda
+load: false
+load_path: ''
+output_log_path: ./trained_models/hmp/amass/autoencoder
+batch_size: 64
+batch_size_eval: 512
+if_run_validation: False
+eval_frequency: 25
+num_epochs: 300
+num_iteration_eval: 10
+num_workers: 4
+seed: 52345
+use_lr_scheduler: true
+lr_scheduler_kwargs:
+  lr_scheduler_type: ExponentialLRSchedulerWarmup
+  warmup_duration: 10
+  update_every: 1
+  min_lr: 0.0001
+  gamma_decay: 0.98
+loss_pose_type: l1
+lr: 0.005
+latent_size: 96
+output_size: 3
+z_activation: tanh
+num_iter_perepoch: 580
+obs_length: 30
+prediction_horizon_train: 120
+prediction_horizon_eval: 120
+prediction_horizon: 120
+pred_length: 120
+prediction_horizon_train_min: 10
+prediction_horizon_train_min_from_epoch: 200
+curriculum_it: 10
+random_prediction_horizon: true
+if_consider_hip: false
+history_sec: 0.5
+prediction_horizon_sec: 2
+motion_repr_type: SkeletonRescalePose
+pose_box_size: 1.2
+seq_centering: 0
+task_name: hmp
+num_joints: 22
+fps: 60
+multimodal_threshold: 0.4
+dataset_type: AMASSDataset
+dataset_name: amass
+precomputed_folder: ./datasets/processed/AMASS/hmp
+annotations_folder: ./datasets/annotations/AMASS/hmp
+dtype: float32
+data_loader_train:
+  stride: 60
+  augmentation: 30
+  shuffle: true
+  datasets:
+  - ACCAD
+  - BMLhandball
+  - BMLmovi
+  - BMLrub
+  - EKUT
+  - CMU
+  - EyesJapanDataset
+  - KIT
+  - PosePrior
+  - TCDHands
+  - TotalCapture
+  da_mirroring: 0.5
+  da_rotations: 1.0
+  drop_last: true
+  if_load_mmgt: false
+data_loader_train_eval:
+  stride: 30
+  augmentation: 0
+  shuffle: false
+  datasets:
+  - ACCAD
+  - BMLhandball
+  - BMLmovi
+  - BMLrub
+  - EKUT
+  - CMU
+  - EyesJapanDataset
+  - KIT
+  - PosePrior
+  - TCDHands
+  - TotalCapture
+  da_mirroring: 0.0
+  da_rotations: 0.0
+  drop_last: false
+  if_load_mmgt: false
+data_loader_valid:
+  stride: 30
+  augmentation: 0
+  shuffle: false
+  datasets:
+  - HumanEva
+  - HDM05
+  - SFU
+  - MoSh
+  file_idces: all
+  drop_last: false
+  if_load_mmgt: false
+
+autoenc_arch:
+  enc_num_layers: 2
+  encoder_hidden_size: 96
+  decoder_hidden_size: 96
+  arch: AutoEncoder 
+  recurrent_arch_enc: StaticGraphGRU
+  recurrent_arch_decoder: StaticGraphGRU
+

trained_models/hmp/amass/diffusion/checkpoints/cvpr_release.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3412e0e99ec21641342a9997b6fdf1f9cd37211cdd79479a43a81ffd1e9820ec
+size 643415447

trained_models/hmp/amass/diffusion/config.yaml

@@ -0,0 +1,56 @@
+if_resume_training: false
+device: cuda
+load: false
+load_path: ''
+dataset_main_path: ./datasets
+dataset_annotation_path: ./datasets/annotations
+dataset_precomputed_path: ./datasets/processed
+output_log_path: ./trained_models/hmp/amass/diffusion
+pretrained_autoencoder_path: ./trained_models/hmp/amass/autoencoder/checkpoints/cvpr_release.pt
+lr: 0.001
+diffusion_objective: pred_x0
+weight_decay: 0.0
+if_use_ema: true
+step_start_ema: 100
+ema_power: 0.6666666666666666
+ema_update_every: 10
+ema_min_value: 0.0
+use_lr_scheduler: true
+lr_scheduler_kwargs:
+  lr_scheduler_type: ExponentialLRSchedulerWarmup
+  warmup_duration: 75
+  update_every: 2
+  min_lr: 0.0001
+  gamma_decay: 0.98
+diffusion_conditioning: true
+num_epochs: 150
+num_workers: 4
+batch_size: 64
+batch_size_eval: 256
+eval_frequency: 25
+train_pick_best_sample_among_k: 50
+similarity_space: input_space
+diffusion_activation: identity
+num_prob_samples: 50
+diffusion_timesteps: 10
+diffusion_type: NonisotropicGaussianDiffusion
+diffusion_loss_type: snr
+loss_reduction_type: l1
+if_run_as_isotropic: false
+if_sigma_n_scale: true
+diffusion_covariance_type: skeleton-diffusion
+gamma_scheduler: cosine
+beta_schedule: cosine
+sigma_n_scale: spectral
+num_iter_perepoch: null
+seed: 63485
+diffusion_arch:
+  arch: Denoiser
+  use_attention: true
+  self_condition: false
+  norm_type: none
+  depth: 4
+  attn_dim_head: 32
+  attn_heads: 8
+  learn_influence: true
+covariance_matrix_type: adjacency

trained_models/hmp/freeman/autoencoder/checkpoints/cvpr_release.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8684cae605f16ec84bbad6e290c44b543fa88d76b39df4a851cbf8a99e9d5f91
+size 15273931

trained_models/hmp/freeman/autoencoder/config.yaml

@@ -0,0 +1,87 @@
+dataset_main_path: ./datasets
+dataset_annotation_path: ./datasets/annotations
+dataset_precomputed_path: ./datasets/processed
+if_resume_training: false
+device: cuda
+load: false
+load_path: ''
+output_log_path: ./trained_models/hmp/freeman/autoencoder
+batch_size: 64
+batch_size_eval: 512
+if_run_validation: False
+eval_frequency: 25
+num_epochs: 200
+num_iteration_eval: 10
+num_workers: 4
+seed: 52345
+use_lr_scheduler: true
+lr_scheduler_kwargs:
+  lr_scheduler_type: ExponentialLRSchedulerWarmup
+  warmup_duration: 10
+  update_every: 1
+  min_lr: 0.0001
+  gamma_decay: 0.98
+loss_pose_type: l1
+lr: 0.005
+latent_size: 96
+output_size: 3
+z_activation: tanh
+num_iter_perepoch: 580
+obs_length: 15
+prediction_horizon_train: 60
+prediction_horizon_eval: 60
+prediction_horizon: 60
+pred_length: 60
+prediction_horizon_train_min: 10
+prediction_horizon_train_min_from_epoch: 200
+curriculum_it: 10
+random_prediction_horizon: true
+if_consider_hip: false
+history_sec: 0.5
+prediction_horizon_sec: 2
+motion_repr_type: SkeletonRescalePose
+pose_box_size: 1.5
+seq_centering: 0
+task_name: hmp
+num_joints: 18
+fps: 30
+multimodal_threshold: 0.5
+dataset_type: FreeManDataset
+dataset_name: freeman
+precomputed_folder: ./datasets/processed/FreeMan/hmp
+annotations_folder: ./datasets/annotations/FreeMan/hmp
+dtype: float32
+data_loader_train:
+  stride: 10
+  augmentation: 5
+  shuffle: true
+  actions: all
+  da_mirroring: 0.5
+  da_rotations: 1.0
+  drop_last: true
+  if_load_mmgt: false
+data_loader_train_eval:
+  stride: 30
+  augmentation: 0
+  shuffle: false
+  actions: all
+  da_mirroring: 0.0
+  da_rotations: 0.0
+  drop_last: false
+  if_load_mmgt: false
+data_loader_valid:
+  stride: 30
+  augmentation: 0
+  shuffle: false
+  segments_path: ./datasets/annotations/FreeMan/hmp/segments_valid.csv
+  actions: all
+  drop_last: false
+  if_load_mmgt: false
+
+autoenc_arch:
+  enc_num_layers: 1
+  encoder_hidden_size: 96
+  decoder_hidden_size: 96
+  arch: AutoEncoder
+  recurrent_arch_enc: StaticGraphGRU
+  recurrent_arch_decoder: StaticGraphGRU

trained_models/hmp/freeman/diffusion/checkpoints/cvpr_release.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:50dcf7bb231d0712f03baba0d761608352b97933c11b6c8f1d4091cc5dfcfa47
+size 466423511

trained_models/hmp/freeman/diffusion/config.yaml

@@ -0,0 +1,56 @@
+if_resume_training: false
+device: cuda
+load: false
+load_path: ''
+dataset_main_path: ./datasets
+dataset_annotation_path: ./datasets/annotations
+dataset_precomputed_path: ./datasets/processed
+output_log_path: ./trained_models/hmp/freeman/diffusion
+pretrained_autoencoder_path: ./trained_models/hmp/freeman/autoencoder/checkpoints/cvpr_release.pt
+lr: 0.001
+diffusion_objective: pred_x0
+weight_decay: 0.0
+if_use_ema: true
+step_start_ema: 100
+ema_power: 0.6666666666666666
+ema_update_every: 10
+ema_min_value: 0.0
+use_lr_scheduler: true
+lr_scheduler_kwargs:
+  lr_scheduler_type: ExponentialLRSchedulerWarmup
+  warmup_duration: 25
+  update_every: 5
+  min_lr: 0.0001
+  gamma_decay: 0.85
+diffusion_conditioning: true
+num_epochs: 200
+num_workers: 4
+batch_size: 64
+batch_size_eval: 256
+eval_frequency: 25
+train_pick_best_sample_among_k: 50
+similarity_space: input_space
+diffusion_activation: identity
+num_prob_samples: 50
+diffusion_timesteps: 10
+diffusion_type: NonisotropicGaussianDiffusion
+diffusion_loss_type: snr
+loss_reduction_type: l1
+if_run_as_isotropic: false
+if_sigma_n_scale: true
+diffusion_covariance_type: skeleton-diffusion
+gamma_scheduler: cosine
+beta_schedule: cosine
+sigma_n_scale: spectral
+num_iter_perepoch: null
+seed: 63485
+diffusion_arch:
+  arch: Denoiser
+  use_attention: true
+  self_condition: false
+  norm_type: none
+  depth: 4
+  attn_dim_head: 32
+  attn_heads: 8
+  learn_influence: true
+covariance_matrix_type: adjacency

trained_models/hmp/h36m/autoencoder/checkpoints/cvpr_release.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:071937a8da740fa99a374a64e09f4ac4ab946a3025da3b7b271bc3bb4a7b3781
+size 16956747

trained_models/hmp/h36m/autoencoder/config.yaml

@@ -0,0 +1,102 @@
+dataset_main_path: ./datasets
+dataset_annotation_path: ./datasets/annotations
+dataset_precomputed_path: ./datasets/processed
+if_resume_training: false
+device: cuda
+load: false
+load_path: ''
+output_log_path: ./trained_models/hmp/h36m/autoencoder
+batch_size: 64
+batch_size_eval: 512
+if_run_validation: false
+eval_frequency: 25
+num_epochs: 200
+num_iteration_eval: 10
+num_workers: 4
+seed: 52345
+use_lr_scheduler: true
+lr_scheduler_kwargs:
+  lr_scheduler_type: ExponentialLRSchedulerWarmup
+  warmup_duration: 10
+  update_every: 1
+  min_lr: 0.0001
+  gamma_decay: 0.98
+loss_pose_type: l1
+lr: 0.005
+latent_size: 96
+output_size: 3
+z_activation: tanh
+num_iter_perepoch: 485
+obs_length: 25
+prediction_horizon_train: 100
+prediction_horizon_eval: 100
+prediction_horizon: 100
+pred_length: 100
+prediction_horizon_train_min: 10
+prediction_horizon_train_min_from_epoch: 200
+curriculum_it: 10
+random_prediction_horizon: true
+if_consider_hip: false
+history_sec: 0.5
+prediction_horizon_sec: 2
+motion_repr_type: SkeletonRescalePose
+pose_box_size: 1.5
+seq_centering: 0
+task_name: hmp
+num_joints: 17
+fps: 50
+multimodal_threshold: 0.5
+dataset_type: H36MDataset
+dataset_name: h36m
+precomputed_folder: ./datasets/processed/Human36M/hmp/
+annotations_folder: ./datasets/annotations/Human36M/hmp
+dtype: float32
+data_loader_train:
+  stride: 10
+  augmentation: 5
+  shuffle: true
+  subjects:
+  - S1
+  - S5
+  - S6
+  - S7
+  - S8
+  actions: all
+  da_mirroring: 0.5
+  da_rotations: 1.0
+  drop_last: true
+  if_load_mmgt: false
+data_loader_train_eval:
+  stride: 30
+  augmentation: 0
+  shuffle: false
+  subjects:
+  - S1
+  - S5
+  - S6
+  - S7
+  - S8
+  actions: all
+  da_mirroring: 0.0
+  da_rotations: 0.0
+  drop_last: false
+  if_load_mmgt: false
+data_loader_valid:
+  stride: 30
+  augmentation: 0
+  shuffle: false
+  subjects:
+  - S8
+  segments_path: ./datasets/annotations/Human36M/hmp/segments_valid.csv
+  actions: all
+  drop_last: false
+  if_load_mmgt: false
+
+
+autoenc_arch:
+  enc_num_layers: 1
+  encoder_hidden_size: 96
+  decoder_hidden_size: 96
+  arch: AutoEncoder 
+  recurrent_arch_enc: StaticGraphGRU
+  recurrent_arch_decoder: StaticGraphGRU

trained_models/hmp/h36m/diffusion/checkpoints/cvpr_release.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:48778f2d8d9dc03a9d25ca4d72a21c491a1d1a7e0ff7707cdacfd81f178cba3d
+size 510561879

trained_models/hmp/h36m/diffusion/config.yaml

@@ -0,0 +1,56 @@
+if_resume_training: false
+device: cuda
+load: false
+load_path: ''
+dataset_main_path: ./datasets
+dataset_annotation_path: ./datasets/annotations
+dataset_precomputed_path: ./datasets/processed
+output_log_path: ./trained_models/hmp/h36m/diffusion
+pretrained_autoencoder_path: ./trained_models/hmp/h36m/autoencoder/checkpoints/cvpr_release.pt
+lr: 0.001
+diffusion_objective: pred_x0
+weight_decay: 0.0
+if_use_ema: true
+step_start_ema: 100
+ema_power: 0.6666666666666666
+ema_update_every: 10
+ema_min_value: 0.0
+use_lr_scheduler: true
+lr_scheduler_kwargs:
+  lr_scheduler_type: ExponentialLRSchedulerWarmup
+  warmup_duration: 25
+  update_every: 2
+  min_lr: 0.0001
+  gamma_decay: 0.85
+diffusion_conditioning: true
+num_epochs: 100
+num_workers: 4
+batch_size: 64
+batch_size_eval: 256
+eval_frequency: 25
+train_pick_best_sample_among_k: 50
+similarity_space: input_space
+diffusion_activation: identity
+num_prob_samples: 50
+diffusion_timesteps: 10
+diffusion_type: NonisotropicGaussianDiffusion
+diffusion_loss_type: snr
+loss_reduction_type: l1
+if_run_as_isotropic: false
+if_sigma_n_scale: true
+diffusion_covariance_type: skeleton-diffusion
+gamma_scheduler: cosine
+beta_schedule: cosine
+sigma_n_scale: spectral
+num_iter_perepoch: null
+seed: 63485
+diffusion_arch:
+  arch: Denoiser
+  use_attention: true
+  self_condition: false
+  norm_type: none
+  depth: 4
+  attn_dim_head: 32
+  attn_heads: 8
+  learn_influence: true
+covariance_matrix_type: adjacency