corrected the construction of occupancy map with index of humans for …

…channel size = 3
chenshengduo · Sep 7, 2019 · 9f3071f · 9f3071f
1 parent d116523
commit 9f3071f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/crowd_nav/policy/multi_human_rl.py b/crowd_nav/policy/multi_human_rl.py
@@ -151,9 +151,9 @@ def build_occupancy_maps(self, human_states):
  dm[2 * int(index)].append(other_vx[i])
  dm[2 * int(index) + 1].append(other_vy[i])
  elif self.om_channel_size == 3:
- dm[2 * int(index)].append(1)
- dm[2 * int(index) + 1].append(other_vx[i])
- dm[2 * int(index) + 2].append(other_vy[i])
+ dm[3 * int(index)].append(1)
+ dm[3 * int(index) + 1].append(other_vx[i])
+ dm[3 * int(index) + 2].append(other_vy[i])
  else:
  raise NotImplementedError
  for i, cell in enumerate(dm):