Skip to content

Commit 6826bda

Browse files
author
Changan
committed
remove attention weights for human colors
1 parent d382e00 commit 6826bda

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

crowd_sim/envs/crowd_sim.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -500,8 +500,8 @@ def render(self, mode='human', output_file=None):
500500

501501
# add humans and their numbers
502502
human_positions = [[state[1][j].position for j in range(len(self.humans))] for state in self.states]
503-
humans = [plt.Circle(human_positions[0][i], self.humans[i].radius, fill=False,
504-
color=(self.attention_weights[0][i], 0, 0)) for i in range(len(self.humans))]
503+
humans = [plt.Circle(human_positions[0][i], self.humans[i].radius, fill=False)
504+
for i in range(len(self.humans))]
505505
human_numbers = [plt.text(humans[i].center[0] - x_offset, humans[i].center[1] - y_offset, str(i),
506506
color='black', fontsize=12) for i in range(len(self.humans))]
507507
for i, human in enumerate(humans):

0 commit comments

Comments
 (0)