bot commited on
Commit
70622cd
·
1 Parent(s): a8eb6e5

Set SO100 normalization in factory code

Browse files
.gitattributes CHANGED
@@ -1,5 +1,5 @@
 
1
  *.webp filter=lfs diff=lfs merge=lfs -text
2
  *.png filter=lfs diff=lfs merge=lfs -text
3
  *.jpg filter=lfs diff=lfs merge=lfs -text
4
  *.safetensors filter=lfs diff=lfs merge=lfs -text
5
- *.bag filter=lfs diff=lfs merge=lfs -text
 
1
+ *.bag filter=lfs diff=lfs merge=lfs -text
2
  *.webp filter=lfs diff=lfs merge=lfs -text
3
  *.png filter=lfs diff=lfs merge=lfs -text
4
  *.jpg filter=lfs diff=lfs merge=lfs -text
5
  *.safetensors filter=lfs diff=lfs merge=lfs -text
 
lerobot/src/lerobot/datasets/factory.py CHANGED
@@ -97,6 +97,14 @@ def make_dataset(cfg: TrainPipelineConfig) -> LeRobotDataset | MultiLeRobotDatas
97
  "observation.images.image": "observation.images.base_0_rgb",
98
  "observation.images.image2": "observation.images.left_wrist_0_rgb",
99
  }
 
 
 
 
 
 
 
 
100
  dataset = SO100Dataset(
101
  data_root=data_root,
102
  index_path=index_path,
 
97
  "observation.images.image": "observation.images.base_0_rgb",
98
  "observation.images.image2": "observation.images.left_wrist_0_rgb",
99
  }
100
+ # Set normalization for SO100 (MEAN_STD for actions/state, IDENTITY for images)
101
+ from lerobot.configs.types import FeatureType, NormalizationMode
102
+ if not cfg.policy.normalization_mapping:
103
+ cfg.policy.normalization_mapping = {
104
+ FeatureType.VISUAL: NormalizationMode.IDENTITY,
105
+ FeatureType.STATE: NormalizationMode.MEAN_STD,
106
+ FeatureType.ACTION: NormalizationMode.MEAN_STD,
107
+ }
108
  dataset = SO100Dataset(
109
  data_root=data_root,
110
  index_path=index_path,