Shuffles the training dataset which is based on slippy map dir order

daniel-j-h · daniel-j-h · commit 6a671a0ed5ed · 2018-07-10T15:02:21.000+02:00
diff --git a/robosat/tools/train.py b/robosat/tools/train.py
@@ -201,7 +201,7 @@ def get_dataset_loaders(model, dataset):
         [os.path.join(path, "validation", "images")], os.path.join(path, "validation", "labels"), transform
     )
 
-    train_loader = DataLoader(train_dataset, batch_size=batch_size, drop_last=True)
-    val_loader = DataLoader(val_dataset, batch_size=batch_size, drop_last=True)
+    train_loader = DataLoader(train_dataset, batch_size=batch_size, shuffle=True, drop_last=True)
+    val_loader = DataLoader(val_dataset, batch_size=batch_size, shuffle=False, drop_last=True)
 
     return train_loader, val_loader

Original file line number	Diff line number	Diff line change
`@@ -201,7 +201,7 @@ def get_dataset_loaders(model, dataset):`
`201`	`201`	`[os.path.join(path, "validation", "images")], os.path.join(path, "validation", "labels"), transform`
`202`	`202`	`)`
`203`	`203`
`204`		`- train_loader = DataLoader(train_dataset, batch_size=batch_size, drop_last=True)`
`205`		`- val_loader = DataLoader(val_dataset, batch_size=batch_size, drop_last=True)`
	`204`	`+ train_loader = DataLoader(train_dataset, batch_size=batch_size, shuffle=True, drop_last=True)`
	`205`	`+ val_loader = DataLoader(val_dataset, batch_size=batch_size, shuffle=False, drop_last=True)`
`206`	`206`
`207`	`207`	`return train_loader, val_loader`