diff --git a/generate_gt_for_training.py b/generate_gt_for_training.py
index e296029..2a2a776 100644
--- a/generate_gt_for_training.py
+++ b/generate_gt_for_training.py
@@ -64,13 +64,17 @@ def pagexml2label(dir_xml,dir_out,type_output,config):
     help="directory where original images will be written as labels.",
     type=click.Path(exists=True, file_okay=False),
 )
-def image_enhancement(dir_imgs, dir_out_images, dir_out_labels):
-    #dir_imgs = './training_data_sample_enhancement/images'
-    #dir_out_images = './training_data_sample_enhancement/images_gt'
-    #dir_out_labels = './training_data_sample_enhancement/labels_gt'
-
+@click.option(
+    "--scales",
+    "-scs",
+    help="json dictionary where the scales are written.",
+    type=click.Path(exists=True, dir_okay=False),
+)
+def image_enhancement(dir_imgs, dir_out_images, dir_out_labels, scales):
     ls_imgs = os.listdir(dir_imgs)
-    ls_scales = [ 0.5, 0.55, 0.6, 0.65, 0.7, 0.75,  0.8, 0.85,  0.9]
+    with open(scales) as f:
+        scale_dict = json.load(f)
+    ls_scales = scale_dict['scales']
 
     for img in tqdm(ls_imgs):
         img_name = img.split('.')[0]
diff --git a/scales_enhancement.json b/scales_enhancement.json
new file mode 100644
index 0000000..58034f0
--- /dev/null
+++ b/scales_enhancement.json
@@ -0,0 +1,3 @@
+{
+    "scales" : [ 0.5, 0.55, 0.6, 0.65, 0.7, 0.75,  0.8, 0.85,  0.9]
+}