allenai / procthor

🏘️ Scaling Embodied AI by Procedurally Generating Interactive 3D Houses
https://procthor.allenai.org/
Apache License 2.0
241 stars 20 forks source link

Where can I get full object list? #16

Open whcpumpkin opened 1 year ago

whcpumpkin commented 1 year ago

Hi! As paper mentioned that ProcTHOR includes 1,633 interactive household objects across 108 categories, where can I get their name? And when navigating in a scene, how can I access to the all object names used in the scene?

mattdeitke commented 1 year ago

Hi @whcpumpkin!

The asset database is available here. I've also done a quick visualization of it here. Note: the later website is not well maintained. It was just for my personal exploration.

To access asset metadata, look at the event that comes after executing each step:

event = controller.step(action="MoveAhead")
print(event.metadata["objects"])

this is the same way one accesses metadata about objects in AI2-THOR, more generally. See the object metadata documentation for more.

whcpumpkin commented 1 year ago

Thank you very much!