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,
|