steffen/hom_traj_gen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

train running

Browse Source
This commit is contained in:
Steffen Illium
2020-03-09 21:41:50 +01:00
parent daed810958
commit 6cc978e464
7 changed files with 68 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lib/visualization/generator_eval.py
View File

@ -19,9 +19,9 @@ class GeneratorVisualizer(object):
def _build_column_dict_list(self):
dict_list = []
for idx in range(self.maps):
image = self.maps[idx] + self.trajectories[idx] + self.generated_alternatives
label = self.labels[idx]
for idx in range(self.maps.shape[0]):
image = (self.maps[idx] + self.trajectories[idx] + self.generated_alternatives[idx]).cpu().numpy().squeeze()
label = int(self.labels[idx])
dict_list.append(dict(image=image, label=label))
half_size = int(len(dict_list) // 2)
return dict_list[:half_size], dict_list[half_size:]
@ -33,10 +33,10 @@ class GeneratorVisualizer(object):
axes_pad=0.2, # pad between axes in inch.
)
for idx in grid.axes_all:
for idx in range(len(grid.axes_all)):
row, col = divmod(idx, len(self.column_dict_list))
current_image = self.column_dict_list[col]['image'][row]
current_label = self.column_dict_list[col]['label'][row]
current_image = self.column_dict_list[col][row]['image']
current_label = self.column_dict_list[col][row]['label']
grid[idx].imshow(current_image)
grid[idx].title.set_text(current_label)
fig.cbar_mode = 'single'