Commit 5fcbb0d
authored
Loads real2sim asset in locomanipulation SDG (#4394)
# Description
This PR adds support to load a USD background in locomanipulation SDG
pipeline.
1. Load NuRec asset in locomanipulation SDG script.
2. Instantiates the first table in the freespace of the background
occupancy map. It resets robot/object position according to table
position. It also writes the default state for robot/object state, so
these states are correct when resetting scene due to termination.
3. Termination conditions:
- Add a termination condition `object_too_far_from_robot`, to check if
robot picks up the wheel.
- Add success termination check on task completion, and write successful
episodes. Use object relative pose to the table for success check.
It was tested by:
Generate locomanipulation SDG data with the
hand_hold-endeavor-wormhole-2 dataset at:
https://huggingface.co/datasets/nvidia/PhysicalAI-Robotics-NuRec/tree/main
./isaaclab.sh -p
scripts/imitation_learning/locomanipulation_sdg/generate_data.py \
--device cpu \
--kit_args="--enable isaacsim.replicator.mobility_gen" \
--task="Isaac-G1-SteeringWheel-Locomanipulation" \
--dataset ./datasets/generated_dataset_g1_locomanip.hdf5 \
--num_runs 1 \
--lift_step 60 \
--navigate_step 130 \
--enable_pinocchio \
--output_file ./datasets/generated_dataset_g1_locomanipulation_sdg.hdf5
\
--enable_cameras \
--background_usd_path <PATH_TO_USD_ASSET>/stage.usdz \
--background_occupancy_yaml_file <PATH_TO_USD_ASSET>/occupancy_map.yaml
## Type of change
- New feature (non-breaking change which adds functionality)
## Screenshots
<img width="1903" height="1046"
alt="locomanipulation_sdg_usd_background1"
src="https://github.com/user-attachments/assets/facb450a-4f6a-4630-a6b6-4e9fa4388825"
/>
<img width="829" height="441" alt="locomanipulation_sdg_usd_background2"
src="https://github.com/user-attachments/assets/b0da6711-f812-4be5-b783-67fd5fd6d741"
/>
## Checklist
- [ ] I have read and understood the [contribution
guidelines](https://isaac-sim.github.io/IsaacLab/main/source/refs/contributing.html)
- [ ] I have run the [`pre-commit` checks](https://pre-commit.com/) with
`./isaaclab.sh --format`
- [ ] I have made corresponding changes to the documentation
- [ ] My changes generate no new warnings
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have updated the changelog and the corresponding version in the
extension's `config/extension.toml` file
- [ ] I have added my name to the `CONTRIBUTORS.md` or my name already
exists there
<!--
As you go through the checklist above, you can mark something as done by
putting an x character in it
For example,
- [x] I have done this task
- [ ] I have not done this task
-->1 parent 97062f7 commit 5fcbb0d
File tree
14 files changed
+875
-157
lines changed- docs/source/overview/imitation-learning
- scripts
- imitation_learning/locomanipulation_sdg
- tools
- source
- isaaclab_mimic/isaaclab_mimic/locomanipulation_sdg
- envs
- isaaclab_tasks/isaaclab_tasks/manager_based/locomanipulation/pick_place
- mdp
- isaaclab/isaaclab/scene
14 files changed
+875
-157
lines changedLines changed: 70 additions & 3 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
408 | 408 | | |
409 | 409 | | |
410 | 410 | | |
| 411 | + | |
| 412 | + | |
411 | 413 | | |
412 | 414 | | |
413 | 415 | | |
| |||
554 | 556 | | |
555 | 557 | | |
556 | 558 | | |
557 | | - | |
| 559 | + | |
| 560 | + | |
| 561 | + | |
558 | 562 | | |
559 | 563 | | |
560 | 564 | | |
561 | 565 | | |
562 | 566 | | |
563 | 567 | | |
564 | 568 | | |
565 | | - | |
566 | | - | |
| 569 | + | |
| 570 | + | |
567 | 571 | | |
568 | 572 | | |
569 | 573 | | |
| |||
590 | 594 | | |
591 | 595 | | |
592 | 596 | | |
| 597 | + | |
| 598 | + | |
| 599 | + | |
| 600 | + | |
| 601 | + | |
| 602 | + | |
| 603 | + | |
| 604 | + | |
| 605 | + | |
| 606 | + | |
| 607 | + | |
| 608 | + | |
| 609 | + | |
| 610 | + | |
| 611 | + | |
| 612 | + | |
| 613 | + | |
| 614 | + | |
| 615 | + | |
| 616 | + | |
| 617 | + | |
| 618 | + | |
| 619 | + | |
| 620 | + | |
| 621 | + | |
| 622 | + | |
| 623 | + | |
| 624 | + | |
| 625 | + | |
| 626 | + | |
| 627 | + | |
| 628 | + | |
| 629 | + | |
| 630 | + | |
| 631 | + | |
| 632 | + | |
| 633 | + | |
| 634 | + | |
| 635 | + | |
| 636 | + | |
| 637 | + | |
| 638 | + | |
| 639 | + | |
| 640 | + | |
| 641 | + | |
| 642 | + | |
| 643 | + | |
| 644 | + | |
| 645 | + | |
| 646 | + | |
| 647 | + | |
| 648 | + | |
| 649 | + | |
| 650 | + | |
| 651 | + | |
| 652 | + | |
| 653 | + | |
| 654 | + | |
| 655 | + | |
| 656 | + | |
| 657 | + | |
| 658 | + | |
| 659 | + | |
0 commit comments