Add UMI-gripper dataset (#83)

Co-authored-by: Remi <re.cadene@gmail.com>
2024-04-28 18:41:07 +02:00
parent a4b6c5e3b1
commit 81e490d46f
11 changed files with 706 additions and 29 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -54,6 +54,7 @@ debugpy = {version = "^1.8.1", optional = true}
 pytest = {version = "^8.1.0", optional = true}
 pytest-cov = {version = "^5.0.0", optional = true}
 datasets = "^2.19.0"
+imagecodecs = { version = "^2024.1.1", optional = true }
 torchaudio = "^2.3.0"


@@ -63,6 +64,7 @@ xarm = ["gym-xarm"]
 aloha = ["gym-aloha"]
 dev = ["pre-commit", "debugpy"]
 test = ["pytest", "pytest-cov"]
+umi = ["imagecodecs"]


 [tool.ruff]