How to create custom dataset for object navigation and rearrangement?

allenai / allenact

An open source framework for research in Embodied-AI from AI2.

https://www.allenact.org

Other

308 stars 49 forks source link

How to create custom dataset for object navigation and rearrangement? #357

Open andykim0723 opened 1 year ago

andykim0723 commented 1 year ago

I am trying to customize existing floorplans. I am trying to change object's color and put new object in the room. I am customizing environment using unity following ai2thor github. I am able to make "FloorPlan_physics.unity" file here but have no clue how to convert it to object navigation/rearrangement dataset. How do I create ".pkl.gz" dataset for rearrangment and ".json.gz" dataset for object navigation?

jordis-ai2 commented 1 year ago

Hi @andykim0723 ,

The dataset generation script in ai2thor-rearrangement shows how to generate datasets for that task. For object navigation, I'd suggest using this script as a starting point. Let me know if this works for you.