Minor edits

Author: ishanrai05

simplifying_rough_sketches/main.py

@@ -31,7 +31,7 @@ def main():
     out = args.out
     create_directory(out)
 
-    root = 'Data'
+    root = args.root
     Input = os.path.join(root, 'Input')
     Target = os.path.join(root,'Target')
     # height = width = 424

simplifying_rough_sketches/utils.py

@@ -63,10 +63,10 @@ def argument_parser():
     parser.add_argument('--height', '-ht', type=int, default=64, help='height of the image to resize to')
     parser.add_argument('--width', '-wd', type=int, default=64, help='width of the image to resize to')
     parser.add_argument('--samples', type=bool, default=False, help='See sample training images')
-    parser.add_argument('--num_epochs', type=int, default=1, help='Number of epochs to train on')
+    parser.add_argument('--num_epochs', type=int, default=75, help='Number of epochs to train on')
     parser.add_argument('--train', default=True, type=bool, help='train the model')
     parser.add_argument('--root', default='.', type=str, help='Root Directory for Input and Target images.')
-    parser.add_argument('--n_folds', default=2, type=int, help='Number of folds in k-fold cross validation.')
+    parser.add_argument('--n_folds', default=7, type=int, help='Number of folds in k-fold cross validation.')
     parser.add_argument('--save_model', default=True, type=bool, help='Save model after training.')
     parser.add_argument('--load_model', default=None, type=str, help='Load existing model.')
     parser.add_argument('--predict', default=None, type=str, help='Path of rough sketch to simplify using existing model')