How to create custom dataset for object navigation and rearrangement?

[andykim0723]

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]

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.