Fixed default settings of CIFAR-10 interface in get_data_generator

cvjena · Jan 31, 2019 · be4d4bb · be4d4bb
1 parent f3c0ab7
commit be4d4bb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/datasets/__init__.py b/datasets/__init__.py
@@ -62,7 +62,8 @@ def get_data_generator(dataset, data_root, classes = None):
 
     if dataset == 'cifar-10':
 
-        return CifarGenerator(data_root, classes, reenumerate = True, cifar10 = True, randzoom_range = 0.25)
+        return CifarGenerator(data_root, classes, reenumerate = True, cifar10 = True,
+                              train_generator_kwargs = { 'horizontal_flip' : True, 'width_shift_range' : 0.15, 'height_shift_range' : 0.15, 'zoom_range' : 0.25 })
 
     elif dataset == 'cifar-100':