Commit 450a4cbf authored by sjjsmuel's avatar sjjsmuel

add augmentation back to training

parent 94796478
......@@ -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()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment