Hugging Face's logo

Beable
/
kkkttt

LeRobot
Model card Files
xet
Community
Beable commited on
Commit
f9ff414
·
verified ·
1 Parent(s): 4980c9d

Upload folder using huggingface_hub

Browse files
Files changed (8) hide show
  1. .gitattributes +23 -0
  2. README.md +158 -0
  3. data/chunk-000/episode_000001.parquet +3 -0
  4. meta/episodes.jsonl +1 -0
  5. meta/episodes_stats.jsonl +1 -0
  6. meta/info.json +118 -0
  7. meta/tasks.jsonl +1 -0
  8. videos/chunk-000/observation.image/episode_000001.mp4 +3 -0
.gitattributes CHANGED
@@ -8,6 +8,7 @@
8
  *.h5 filter=lfs diff=lfs merge=lfs -text
9
  *.joblib filter=lfs diff=lfs merge=lfs -text
10
  *.lfs.* filter=lfs diff=lfs merge=lfs -text
 
11
  *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
  *.model filter=lfs diff=lfs merge=lfs -text
13
  *.msgpack filter=lfs diff=lfs merge=lfs -text
@@ -33,3 +34,25 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  *.h5 filter=lfs diff=lfs merge=lfs -text
9
  *.joblib filter=lfs diff=lfs merge=lfs -text
10
  *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.lz4 filter=lfs diff=lfs merge=lfs -text
12
  *.mlmodel filter=lfs diff=lfs merge=lfs -text
13
  *.model filter=lfs diff=lfs merge=lfs -text
14
  *.msgpack filter=lfs diff=lfs merge=lfs -text
 
34
  *.zip filter=lfs diff=lfs merge=lfs -text
35
  *.zst filter=lfs diff=lfs merge=lfs -text
36
  *tfevents* filter=lfs diff=lfs merge=lfs -text
37
+ # Audio files - uncompressed
38
+ *.pcm filter=lfs diff=lfs merge=lfs -text
39
+ *.sam filter=lfs diff=lfs merge=lfs -text
40
+ *.raw filter=lfs diff=lfs merge=lfs -text
41
+ # Audio files - compressed
42
+ *.aac filter=lfs diff=lfs merge=lfs -text
43
+ *.flac filter=lfs diff=lfs merge=lfs -text
44
+ *.mp3 filter=lfs diff=lfs merge=lfs -text
45
+ *.ogg filter=lfs diff=lfs merge=lfs -text
46
+ *.wav filter=lfs diff=lfs merge=lfs -text
47
+ # Image files - uncompressed
48
+ *.bmp filter=lfs diff=lfs merge=lfs -text
49
+ *.gif filter=lfs diff=lfs merge=lfs -text
50
+ *.png filter=lfs diff=lfs merge=lfs -text
51
+ *.tiff filter=lfs diff=lfs merge=lfs -text
52
+ # Image files - compressed
53
+ *.jpg filter=lfs diff=lfs merge=lfs -text
54
+ *.jpeg filter=lfs diff=lfs merge=lfs -text
55
+ *.webp filter=lfs diff=lfs merge=lfs -text
56
+ # Video files - compressed
57
+ *.mp4 filter=lfs diff=lfs merge=lfs -text
58
+ *.webm filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,158 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ task_categories:
4
+ - robotics
5
+ tags:
6
+ - LeRobot
7
+ configs:
8
+ - config_name: default
9
+ data_files: data/*/*.parquet
10
+ ---
11
+
12
+ This dataset was created using [LeRobot](https://github.com/huggingface/lerobot).
13
+
14
+ ## Dataset Description
15
+
16
+
17
+
18
+ - **Homepage:** https://diffusion-policy.cs.columbia.edu/
19
+ - **Paper:** https://arxiv.org/abs/2303.04137v5
20
+ - **License:** mit
21
+
22
+ ## Dataset Structure
23
+
24
+ [meta/info.json](meta/info.json):
25
+ ```json
26
+ {
27
+ "codebase_version": "v2.0",
28
+ "robot_type": "unknown",
29
+ "total_episodes": 206,
30
+ "total_frames": 25650,
31
+ "total_tasks": 1,
32
+ "total_videos": 206,
33
+ "total_chunks": 1,
34
+ "chunks_size": 1000,
35
+ "fps": 10,
36
+ "splits": {
37
+ "train": "0:206"
38
+ },
39
+ "data_path": "data/chunk-{episode_chunk:03d}/episode_{episode_index:06d}.parquet",
40
+ "video_path": "videos/chunk-{episode_chunk:03d}/{video_key}/episode_{episode_index:06d}.mp4",
41
+ "features": {
42
+ "observation.image": {
43
+ "dtype": "video",
44
+ "shape": [
45
+ 96,
46
+ 96,
47
+ 3
48
+ ],
49
+ "names": [
50
+ "height",
51
+ "width",
52
+ "channel"
53
+ ],
54
+ "video_info": {
55
+ "video.fps": 10.0,
56
+ "video.codec": "av1",
57
+ "video.pix_fmt": "yuv420p",
58
+ "video.is_depth_map": false,
59
+ "has_audio": false
60
+ }
61
+ },
62
+ "observation.state": {
63
+ "dtype": "float32",
64
+ "shape": [
65
+ 2
66
+ ],
67
+ "names": {
68
+ "motors": [
69
+ "motor_0",
70
+ "motor_1"
71
+ ]
72
+ }
73
+ },
74
+ "action": {
75
+ "dtype": "float32",
76
+ "shape": [
77
+ 2
78
+ ],
79
+ "names": {
80
+ "motors": [
81
+ "motor_0",
82
+ "motor_1"
83
+ ]
84
+ }
85
+ },
86
+ "episode_index": {
87
+ "dtype": "int64",
88
+ "shape": [
89
+ 1
90
+ ],
91
+ "names": null
92
+ },
93
+ "frame_index": {
94
+ "dtype": "int64",
95
+ "shape": [
96
+ 1
97
+ ],
98
+ "names": null
99
+ },
100
+ "timestamp": {
101
+ "dtype": "float32",
102
+ "shape": [
103
+ 1
104
+ ],
105
+ "names": null
106
+ },
107
+ "next.reward": {
108
+ "dtype": "float32",
109
+ "shape": [
110
+ 1
111
+ ],
112
+ "names": null
113
+ },
114
+ "next.done": {
115
+ "dtype": "bool",
116
+ "shape": [
117
+ 1
118
+ ],
119
+ "names": null
120
+ },
121
+ "next.success": {
122
+ "dtype": "bool",
123
+ "shape": [
124
+ 1
125
+ ],
126
+ "names": null
127
+ },
128
+ "index": {
129
+ "dtype": "int64",
130
+ "shape": [
131
+ 1
132
+ ],
133
+ "names": null
134
+ },
135
+ "task_index": {
136
+ "dtype": "int64",
137
+ "shape": [
138
+ 1
139
+ ],
140
+ "names": null
141
+ }
142
+ }
143
+ }
144
+ ```
145
+
146
+
147
+ ## Citation
148
+
149
+ **BibTeX:**
150
+
151
+ ```bibtex
152
+ @article{chi2024diffusionpolicy,
153
+ author = {Cheng Chi and Zhenjia Xu and Siyuan Feng and Eric Cousineau and Yilun Du and Benjamin Burchfiel and Russ Tedrake and Shuran Song},
154
+ title ={Diffusion Policy: Visuomotor Policy Learning via Action Diffusion},
155
+ journal = {The International Journal of Robotics Research},
156
+ year = {2024},
157
+ }
158
+ ```
data/chunk-000/episode_000001.parquet ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f80c765d677766f866c88e1d60fd3e74f2f54535f302ea933de803471677fd28
3
+ size 21657
meta/episodes.jsonl ADDED
@@ -0,0 +1 @@
 
 
1
+ {"episode_index": 0, "tasks": ["Push the T-shaped block onto the T-shaped target."], "length": 644}
meta/episodes_stats.jsonl ADDED
@@ -0,0 +1 @@
 
 
1
+ {"episode_index": 0, "stats": {"observation.image": {"min": [[[0.0]], [[0.0]], [[0.0]]], "max": [[[1.0]], [[1.0]], [[1.0]]], "mean": [[[0.971221220462866]], [[0.9597391613774777]], [[0.9511610388548233]]], "std": [[[0.07689539518206108]], [[0.17204725742909527]], [[0.1628096548935342]]], "count": [100]}, "observation.state": {"min": [-0.09973146766424179, 0.41585952043533325], "max": [0.12305367738008499, 0.7031053900718689], "mean": [0.024191024302765596, 0.6012363766758447], "std": [0.06816179310538653, 0.09104716903689453], "count": [427]}, "action": {"min": [-1.0, -1.0], "max": [1.0, 0.8349109292030334], "mean": [0.018332271696563453, -0.0782456231885317], "std": [0.40100337026607946, 0.37188337295521534], "count": [427]}, "episode_index": {"min": [0], "max": [0], "mean": [0.0], "std": [0.0], "count": [427]}, "frame_index": {"min": [0], "max": [426], "mean": [213.0], "std": [123.26394444443193], "count": [427]}, "timestamp": {"min": [0.0], "max": [42.600000000000335], "mean": [21.300000000000086], "std": [12.326394444443299], "count": [427]}, "next.reward": {"min": [0.3751977222397975], "max": [0.9231255931667194], "mean": [0.5822728259444178], "std": [0.1781399972564734], "count": [427]}, "next.done": {"min": [false], "max": [true], "mean": [0.00234192037470726], "std": [0.04833669189824425], "count": [427]}, "next.success": {"min": [false], "max": [true], "mean": [0.00234192037470726], "std": [0.04833669189824425], "count": [427]}, "index": {"min": [0], "max": [426], "mean": [213.0], "std": [123.26394444443193], "count": [427]}, "task_index": {"min": [0], "max": [0], "mean": [0.0], "std": [0.0], "count": [427]}}}
meta/info.json ADDED
@@ -0,0 +1,118 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "codebase_version": "v2.1",
3
+ "robot_type": "unknown",
4
+ "total_episodes": 1,
5
+ "total_frames": 300,
6
+ "total_tasks": 1,
7
+ "total_videos": 1,
8
+ "total_chunks": 1,
9
+ "chunks_size": 1000,
10
+ "fps": 10,
11
+ "splits": {
12
+ "train": "0:4"
13
+ },
14
+ "data_path": "data/chunk-{episode_chunk:03d}/episode_{episode_index:06d}.parquet",
15
+ "video_path": "videos/chunk-{episode_chunk:03d}/{video_key}/episode_{episode_index:06d}.mp4",
16
+ "features": {
17
+ "observation.image": {
18
+ "dtype": "video",
19
+ "shape": [
20
+ 302,
21
+ 362,
22
+ 3
23
+ ],
24
+ "names": [
25
+ "height",
26
+ "width",
27
+ "channel"
28
+ ],
29
+ "video_info": {
30
+ "video.fps": 10.0,
31
+ "video.codec": "av1",
32
+ "video.pix_fmt": "yuv420p",
33
+ "video.is_depth_map": false,
34
+ "has_audio": false
35
+ }
36
+ },
37
+ "observation.state": {
38
+ "dtype": "float32",
39
+ "shape": [
40
+ 2
41
+ ],
42
+ "names": {
43
+ "motors": [
44
+ "motor_0",
45
+ "motor_1"
46
+ ]
47
+ }
48
+ },
49
+ "action": {
50
+ "dtype": "float32",
51
+ "shape": [
52
+ 2
53
+ ],
54
+ "names": {
55
+ "motors": [
56
+ "motor_0",
57
+ "motor_1"
58
+ ]
59
+ }
60
+ },
61
+ "episode_index": {
62
+ "dtype": "int64",
63
+ "shape": [
64
+ 1
65
+ ],
66
+ "names": null
67
+ },
68
+ "frame_index": {
69
+ "dtype": "int64",
70
+ "shape": [
71
+ 1
72
+ ],
73
+ "names": null
74
+ },
75
+ "timestamp": {
76
+ "dtype": "float32",
77
+ "shape": [
78
+ 1
79
+ ],
80
+ "names": null
81
+ },
82
+ "next.reward": {
83
+ "dtype": "float32",
84
+ "shape": [
85
+ 1
86
+ ],
87
+ "names": null
88
+ },
89
+ "next.done": {
90
+ "dtype": "bool",
91
+ "shape": [
92
+ 1
93
+ ],
94
+ "names": null
95
+ },
96
+ "next.success": {
97
+ "dtype": "bool",
98
+ "shape": [
99
+ 1
100
+ ],
101
+ "names": null
102
+ },
103
+ "index": {
104
+ "dtype": "int64",
105
+ "shape": [
106
+ 1
107
+ ],
108
+ "names": null
109
+ },
110
+ "task_index": {
111
+ "dtype": "int64",
112
+ "shape": [
113
+ 1
114
+ ],
115
+ "names": null
116
+ }
117
+ }
118
+ }
meta/tasks.jsonl ADDED
@@ -0,0 +1 @@
 
 
1
+ {"task_index": 0, "task": "Push the T-shaped block onto the T-shaped target."}
videos/chunk-000/observation.image/episode_000001.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88e61a9674f4db98bd4575e53e5ce3de05dcfdf53708c55ff7edacb367a6e84a
3
+ size 395682