pepijn223 HF Staff commited on
Commit
fd60fdd
·
verified ·
1 Parent(s): 8aa46b4

Use standard LeRobot normalization processors

Browse files
policy_postprocessor.json CHANGED
@@ -5,6 +5,25 @@
5
  "registry_name": "being_h05_semantic_unpack",
6
  "config": {}
7
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  {
9
  "registry_name": "device_processor",
10
  "config": {
@@ -13,4 +32,4 @@
13
  }
14
  }
15
  ]
16
- }
 
5
  "registry_name": "being_h05_semantic_unpack",
6
  "config": {}
7
  },
8
+ {
9
+ "registry_name": "unnormalizer_processor",
10
+ "config": {
11
+ "eps": 1e-08,
12
+ "features": {
13
+ "action": {
14
+ "type": "ACTION",
15
+ "shape": [
16
+ 12
17
+ ]
18
+ }
19
+ },
20
+ "norm_map": {
21
+ "VISUAL": "IDENTITY",
22
+ "STATE": "IDENTITY",
23
+ "ACTION": "IDENTITY"
24
+ }
25
+ }
26
+ },
27
  {
28
  "registry_name": "device_processor",
29
  "config": {
 
32
  }
33
  }
34
  ]
35
+ }
policy_preprocessor.json CHANGED
@@ -5,6 +5,79 @@
5
  "registry_name": "to_batch_processor",
6
  "config": {}
7
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  {
9
  "registry_name": "being_h05_semantic_pack",
10
  "config": {
@@ -69,4 +142,4 @@
69
  }
70
  }
71
  ]
72
- }
 
5
  "registry_name": "to_batch_processor",
6
  "config": {}
7
  },
8
+ {
9
+ "registry_name": "normalizer_processor",
10
+ "config": {
11
+ "eps": 1e-08,
12
+ "features": {
13
+ "observation.images.robot0_agentview_left": {
14
+ "type": "VISUAL",
15
+ "shape": [
16
+ 3,
17
+ 224,
18
+ 224
19
+ ]
20
+ },
21
+ "observation.images.robot0_agentview_right": {
22
+ "type": "VISUAL",
23
+ "shape": [
24
+ 3,
25
+ 224,
26
+ 224
27
+ ]
28
+ },
29
+ "observation.images.robot0_eye_in_hand": {
30
+ "type": "VISUAL",
31
+ "shape": [
32
+ 3,
33
+ 224,
34
+ 224
35
+ ]
36
+ },
37
+ "observation.state.eef_position": {
38
+ "type": "STATE",
39
+ "shape": [
40
+ 3
41
+ ]
42
+ },
43
+ "observation.state.eef_rotation": {
44
+ "type": "STATE",
45
+ "shape": [
46
+ 3
47
+ ]
48
+ },
49
+ "observation.state.gripper_qpos": {
50
+ "type": "STATE",
51
+ "shape": [
52
+ 2
53
+ ]
54
+ },
55
+ "observation.state.base_position": {
56
+ "type": "STATE",
57
+ "shape": [
58
+ 3
59
+ ]
60
+ },
61
+ "observation.state.base_rotation": {
62
+ "type": "STATE",
63
+ "shape": [
64
+ 3
65
+ ]
66
+ },
67
+ "action": {
68
+ "type": "ACTION",
69
+ "shape": [
70
+ 12
71
+ ]
72
+ }
73
+ },
74
+ "norm_map": {
75
+ "VISUAL": "IDENTITY",
76
+ "STATE": "IDENTITY",
77
+ "ACTION": "IDENTITY"
78
+ }
79
+ }
80
+ },
81
  {
82
  "registry_name": "being_h05_semantic_pack",
83
  "config": {
 
142
  }
143
  }
144
  ]
145
+ }