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

Debugging Validation and testing

Browse Source
This commit is contained in:
Si11ium
2020-03-10 08:40:40 +01:00
parent 48f4d84bf3
commit 21e7e31805
3 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
datasets/trajectory_dataset.py
View File

@@ -57,11 +57,11 @@ class TrajDataset(Dataset):
if self.mode.lower() in ['all_in_map', 'separated_arrays']:
map_array = torch.as_tensor(self.map.as_array).float()
if self.mode == 'separated_arrays':
return (map_array, trajectory.draw_in_array(self.map_shape), int(label)), \
alternative.draw_in_array(self.map_shape)
return (map_array, torch.as_tensor(trajectory.draw_in_array(self.map_shape)).float(), int(label)), \
torch.as_tensor(alternative.draw_in_array(self.map_shape)).float()
else:
return torch.cat((map_array, trajectory.draw_in_array(self.map_shape),
alternative.draw_in_array(self.map_shape))), int(label)
return torch.cat((map_array, torch.as_tensor(trajectory.draw_in_array(self.map_shape)).float(),
torch.as_tensor(alternative.draw_in_array(self.map_shape)).float())), int(label)
elif self.mode == 'vectors':
return trajectory.vertices, alternative.vertices, label, self.mapname