Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

YCB v1.2 #1772

Merged
merged 8 commits into from
Jun 10, 2022
Merged
Show file tree
Hide file tree
Changes from 3 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
35 changes: 35 additions & 0 deletions DATASETS.md
Original file line number Diff line number Diff line change
Expand Up @@ -154,6 +154,41 @@ The exported `*.glb` files can directly be used with Habitat-Sim versions >= 2.0

Note: Depending on the configured radius and height of the agent, certain scans may have no navigable locations on the navmesh (~200). These scenes can be filtered out by checking if `sim.pathfinder.is_loaded` is False.

## YCB Benchmarks - Object and Model Set
Details: [https://www.ycbbenchmarks.com/](https://www.ycbbenchmarks.com/).

> YCB Object and Model Set is designed for facilitating benchmarking in robotic manipulation... The set is associated with a [model database](http://www.ycbbenchmarks.com/object-models/) which provides mesh models and high-resolution RGB-D scans of the objects for easy incorporation into manipulation and planning software platforms.

Pre-processed, [Habitat-ready assets](https://dl.fbaipublicfiles.com/habitat/ycb/hab_ycb_v1.2.zip).

Quick-start with the dataset_downloader utility:

```
# with conda install
python -m habitat_sim.utils.datasets_download --uids ycb --data-path /path/to/data/

# with source
python /path/to/habitat_sim/src_python/habitat_sim/utils/datasets_download.py --uids ycb --data-path /path/to/data/
```

Load the assets in python by setting [MetadataMediator.active_dataset](https://aihabitat.org/docs/habitat-sim/habitat_sim.metadata.MetadataMediator.html#active_dataset):
```
#load the full YCB dataset into the MetadataMediator
sim.metadata_mediator.active_dataset = "/path/to/data/objects/ycb/ycb.scene_dataset_config.json"

#then instance objects from the ObjectAttributesManager:
chef_can_key = sim.get_object_template_manager().get_file_template_handles("002_master_chef_can")[0]
chef_can_object = sim.get_rigid_object_manager().add_object_by_template_handle(chef_can_key)
```
For more information on using objects in habitat-sim, see the "Habitat-Sim for Interaction" and "Habitat-Sim Advanced Topics" sections of our [ECCV tutorial series](https://aihabitat.org/tutorial/2020/).

To quickly test in the viewer application:
```
#from the habitat-sim directory
./build/viewer --stage-requires-lighting --enable-physics --object-dir "" --dataset data/objects/ycb/ycb.scene_dataset_config.json -- data/test_assets/scenes/simple_room.glb
aclegg3 marked this conversation as resolved.
Show resolved Hide resolved
```
Then press `'o'` key to add random objects from the dataset.

# Previewing dataset assets using Habitat-Sim's viewers

For datasets with scene dataset configuration support (such as HM3D, ReplicaCAD, MP3D, Gibson, etc) you can preview the assets using one of Habitat's command-line driven viewers, either in c++ or python. When launching the viewer, you should specify not only the desired scene to load, but also the specifying the scene dataset configuration file, to guarantee the assets load and display correctly. This has the added benefit of providing quick access to other scenes in the same dataset, without requiring a reload of the entire simulation environment from the command line.
Expand Down
6 changes: 3 additions & 3 deletions src_python/habitat_sim/utils/datasets_download.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,10 +121,10 @@ def initialize_test_data_sources(data_path):
"version": "1.5",
},
"ycb": {
"source": "https://dl.fbaipublicfiles.com/habitat/ycb/hab_ycb_v1.1.zip",
"package_name": "hab_ycb_v1.1.zip",
"source": "https://dl.fbaipublicfiles.com/habitat/ycb/hab_ycb_v1.2.zip",
"package_name": "hab_ycb_v1.2.zip",
"link": data_path + "objects/ycb",
"version": "1.1",
"version": "1.2",
},
"hab_fetch": {
"source": "http://dl.fbaipublicfiles.com/habitat/hab_fetch_v1.0.zip",
Expand Down