add augmentation back to training

train_classifier.py

@@ -82,8 +82,7 @@ train_loader = DataLoader(data_path=train_dir,
                           img_width=img_width,
                           img_height=img_height,
                           channels=channels,
-                          # TODO reverse augment to True after testing
-                          augment=False,
+                          augment=True,
                           annotation=annot_loader
                           )
 train_dataset = train_loader.load_dataset()
@@ -96,8 +95,7 @@ validation_loader =  DataLoader(data_path=validation_dir,
                                 img_height=img_height,
                                 channels=channels,
                                 #split_size=options.split_size,
-                                # TODO reverse augment to True after testing
-                                augment=False
+                                augment=True
                                 )
 validation_dataset = validation_loader.load_dataset()