wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix typos in `lerobot/scripts/visualize_dataset.py` (#656)

This commit is contained in:
HUANG TZU-CHUN 2025-01-28 20:07:10 +08:00 committed by GitHub
parent 4def6d6ac2
commit 5daa45436d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lerobot/scripts/visualize_dataset.py
View File

@ -15,14 +15,14 @@
# limitations under the License. # limitations under the License.
""" Visualize data of **all** frames of any episode of a dataset of type LeRobotDataset. """ Visualize data of **all** frames of any episode of a dataset of type LeRobotDataset.
Note: The last frame of the episode doesnt always correspond to a final state. Note: The last frame of the episode doesn't always correspond to a final state.
That's because our datasets are composed of transition from state to state up to That's because our datasets are composed of transition from state to state up to
the antepenultimate state associated to the ultimate action to arrive in the final state. the antepenultimate state associated to the ultimate action to arrive in the final state.
However, there might not be a transition from a final state to another state. However, there might not be a transition from a final state to another state.
Note: This script aims to visualize the data used to train the neural networks. Note: This script aims to visualize the data used to train the neural networks.
~What you see is what you get~. When visualizing image modality, it is often expected to observe ~What you see is what you get~. When visualizing image modality, it is often expected to observe
lossly compression artifacts since these images have been decoded from compressed mp4 videos to lossy compression artifacts since these images have been decoded from compressed mp4 videos to
save disk space. The compression factor applied has been tuned to not affect success rate. save disk space. The compression factor applied has been tuned to not affect success rate.
Examples: Examples:
@ -199,7 +199,7 @@ def main():
"--repo-id", "--repo-id",
type=str, type=str,
required=True, required=True,
help="Name of hugging face repositery containing a LeRobotDataset dataset (e.g. `lerobot/pusht`).", help="Name of hugging face repository containing a LeRobotDataset dataset (e.g. `lerobot/pusht`).",
) )
parser.add_argument( parser.add_argument(
"--episode-index", "--episode-index",