facebookresearch / habitat-lab

A modular high-level library to train embodied AI agents across a variety of tasks and environments.
https://aihabitat.org/
MIT License
1.81k stars 460 forks source link

Map agent position to room_id in Rearrange Episode #1974

Open S4b3 opened 1 month ago

S4b3 commented 1 month ago

Habitat-Lab and Habitat-Sim versions

Habitat-Lab: master (8785790b2b1c53fb4fed3a38939824365ae986ba)

Habitat-Sim: master

❓ Questions and Help

Is there a way to map coordinates to the belonging room during the training process?

I have been trying to extract the generic location of an agent for quite a bit now, and although I can see that RoomGoal and ObjectGoal both have room_id and room_name parameters, thus making it possible to build an instruction such as "Find the mug in the kitchen", i am struggling to find a way to map any set of coordinates to the corresponding room in the episode map.

Is there a way to do this? Let's say my goal would be to have a metric that registers the humanoid agent location at any step e.g. agent_0 is in the living room (_or roomid k) at step 200. Also I've searched for an explicit mapping between room_id and room_name in the docs, but I haven't found it. Is there an existing one?

Thank you!