Added v0.2 dataset links.

Mr-tooth · Apr 3, 2024 · fbf96db · fbf96db
1 parent 5fc0e39
commit fbf96db
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/loco_mujoco/datasets/data_generation/conversion/convert_UnitreeH1_run.py b/loco_mujoco/datasets/data_generation/conversion/convert_UnitreeH1_run.py
@@ -59,6 +59,6 @@
 
  unavailable_keys = []
  dataset = adapt_mocap(path_mat, joint_conf=joint_conf, unavailable_keys=unavailable_keys, rename_map=rename_map,
- discard_first=25000, discard_last=1000)
+ discard_first=28500, discard_last=12500)
 
  np.savez(file=target_path, **dataset)
diff --git a/loco_mujoco/utils/dataset.py b/loco_mujoco/utils/dataset.py
@@ -31,10 +31,10 @@ def download_real_datasets():
  dataset_path_humanoid = dataset_path / "humanoids/real"
  dataset_path_humanoid_str = str(dataset_path_humanoid)
  os.makedirs(dataset_path_humanoid_str, exist_ok=True)
- humanoid_url = "https://zenodo.org/records/10102870/files/humanoid_datasets_v0.1.zip?download=1"
+ humanoid_url = "https://zenodo.org/records/10912641/files/humanoid_datasets_v0.2.zip?download=1"
  wget.download(humanoid_url, out=dataset_path_humanoid_str)
  print("\n")
- file_name = "humanoid_datasets_v0.1.zip"
+ file_name = "humanoid_datasets_v0.2.zip"
  file_path = str(dataset_path_humanoid / file_name)
  with zipfile.ZipFile(file_path, "r") as zip_ref:
  zip_ref.extractall(dataset_path_humanoid_str)
@@ -44,10 +44,10 @@ def download_real_datasets():
  dataset_path_quadrupeds = dataset_path / "quadrupeds/real"
  dataset_path_quadrupeds_str = str(dataset_path_quadrupeds)
  os.makedirs(dataset_path_quadrupeds_str, exist_ok=True)
- quadruped_url = "https://zenodo.org/records/10102870/files/quadruped_datasets_v0.1.zip?download=1"
+ quadruped_url = "https://zenodo.org/records/10912641/files/quadruped_datasets_v0.2.zip?download=1"
  wget.download(quadruped_url, out=dataset_path_quadrupeds_str)
  print("\n")
- file_name = "quadruped_datasets_v0.1.zip"
+ file_name = "quadruped_datasets_v0.2.zip"
  file_path = str(dataset_path_quadrupeds / file_name)
  with zipfile.ZipFile(file_path, "r") as zip_ref:
  zip_ref.extractall(dataset_path_quadrupeds_str)
@@ -65,9 +65,9 @@ def download_perfect_datasets():
  dataset_path_humanoid = dataset_path / "humanoids/perfect"
  dataset_path_humanoid_str = str(dataset_path_humanoid)
  os.makedirs(dataset_path_humanoid_str, exist_ok=True)
- humanoid_url = "https://zenodo.org/records/10393490/files/humanoid_datasets_perfect_v0.1.zip?download=1"
+ humanoid_url = "https://zenodo.org/records/10912641/files/humanoid_datasets_perfect_v0.2.zip?download=1"
  wget.download(humanoid_url, out=dataset_path_humanoid_str)
- file_name = "humanoid_datasets_perfect_v0.1.zip"
+ file_name = "humanoid_datasets_perfect_v0.2.zip"
  file_path = str(dataset_path_humanoid / file_name)
  with zipfile.ZipFile(file_path, "r") as zip_ref:
  zip_ref.extractall(dataset_path_humanoid_str)
@@ -78,10 +78,10 @@ def download_perfect_datasets():
  dataset_path_quadrupeds = dataset_path / "quadrupeds/perfect"
  dataset_path_quadrupeds_str = str(dataset_path_quadrupeds)
  os.makedirs(dataset_path_quadrupeds_str, exist_ok=True)
- quadruped_url = "https://zenodo.org/records/10393490/files/quadruped_datasets_perfect_v0.1.zip?download=1"
+ quadruped_url = "https://zenodo.org/records/10912641/files/quadruped_datasets_perfect_v0.2.zip?download=1"
  wget.download(quadruped_url, out=dataset_path_quadrupeds_str)
  print("\n")
- file_name = "quadruped_datasets_perfect_v0.1.zip"
+ file_name = "quadruped_datasets_perfect_v0.2.zip"
  file_path = str(dataset_path_quadrupeds / file_name)
  with zipfile.ZipFile(file_path, "r") as zip_ref:
  zip_ref.extractall(dataset_path_quadrupeds_str)