remove attention weights for human colors

chenshengduo · Oct 9, 2018 · 6826bda · 6826bda
1 parent d382e00
commit 6826bda
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/crowd_sim/envs/crowd_sim.py b/crowd_sim/envs/crowd_sim.py
@@ -500,8 +500,8 @@ def render(self, mode='human', output_file=None):
 
  # add humans and their numbers
  human_positions = [[state[1][j].position for j in range(len(self.humans))] for state in self.states]
- humans = [plt.Circle(human_positions[0][i], self.humans[i].radius, fill=False,
-  color=(self.attention_weights[0][i], 0, 0)) for i in range(len(self.humans))]
+ humans = [plt.Circle(human_positions[0][i], self.humans[i].radius, fill=False)
+ for i in range(len(self.humans))]
  human_numbers = [plt.text(humans[i].center[0] - x_offset, humans[i].center[1] - y_offset, str(i),
  color='black', fontsize=12) for i in range(len(self.humans))]
  for i, human in enumerate(humans):