validation written

2020-03-09 17:17:43 +01:00
parent e7ccfb7947
commit 4ae333fe5d
5 changed files with 133 additions and 24 deletions
--- a/datasets/trajectory_dataset.py
+++ b/datasets/trajectory_dataset.py
@@ -102,7 +102,7 @@ class TrajData(object):

    def _load_datasets(self):
        map_files = list(self.maps_root.glob('*.bmp'))
-        equal_split = int(self.length // len(map_files))
+        equal_split = int(self.length // len(map_files)) or 1

        # find max image size among available maps:
        max_map_size = (1, ) + tuple(reversed(tuple(map(max, *[Image.open(map_file).size for map_file in map_files]))))