search

allenai / ai2thor

An open-source platform for Visual AI.
http://ai2thor.allenai.org
Apache License 2.0
1.17k stars 217 forks source link

Object placement control #300

Open YunlianMoon opened 5 years ago

YunlianMoon commented 5 years ago

Can I control the placement relationship between objects? for example, mug or cup next to coffee machine， apple next to bread, book put on the desk, etc.

winthos commented 5 years ago

We don't currently have support for specifying placement relationship parameters like that in our random spawner. You could try something like use a ton of random seeds and filter out ones that specifically have objects placed close together(ie: only use seeds where the distance from an Apple to Bread is within a threshold).

YunlianMoon commented 5 years ago

@winthos Do you mean to change the value of "randomSeed" in [controller.step(dict(action = 'InitialRandomSpawn', randomSeed = 0)] ？

winthos commented 5 years ago

Yes, if you pass in different seeds, the scene will initialize with objects in new random positions based on the seed.