This commit is contained in:
fxia22 2019-05-07 14:30:32 -07:00
parent 1fceee8b01
commit 2471fabfbd
3 changed files with 6 additions and 3 deletions

View File

@ -58,6 +58,9 @@ Gibson v2 can be installed as a python package:
```bash
git clone https://github.com/fxia22/gibsonv2
cd gibsonv2
conda create -n py3-gibson python=3.6 anaconda
source activate py3-gibson
pip install -e .
```

View File

@ -181,7 +181,7 @@ class NavigateEnv(BaseEnv):
return additional_states
"""
def get_state(self):
def get_state(self, collision_links=[]):
# calculate state
# sensor_state = self.robots[0].calc_state()
# sensor_state = np.concatenate((sensor_state, self.get_additional_states()))
@ -265,7 +265,7 @@ class NavigateEnv(BaseEnv):
def step(self, action):
self.robots[0].apply_action(action)
collision_links = self.run_simulation()
state = self.get_state()
state = self.get_state(collision_links)
reward = self.get_reward(collision_links)
done = self.get_termination()

View File

@ -85,7 +85,7 @@ setup(name='gibson2',
'numpy>=1.13',
'pybullet==2.4.1',
'transforms3d>=0.3.1',
'tqdm >= 4',
'tqdm == 4.19.9',
'Pillow>=4.2.1',
'matplotlib>=2.1.0',
'mpi4py>=2.0.0',