.

1 files changed, 59 insertions, 0 deletions

diff --git a/megapixels/commands/misc/compare_sres.py b/megapixels/commands/misc/compare_sres.py
new file mode 100644
index 00000000..b96570fe
--- /dev/null
+++ b/megapixels/commands/misc/compare_sres.py
@@ -0,0 +1,59 @@
+import click
+
+
+@click.command()
+@click.option('-i', '--orig', 'opt_dir_in_orig', required=True,
+  help='Input directory')
+@click.option('-n', '--new', 'opt_dir_in_new', required=True,
+  help='Input directory files to compare to')
+@click.pass_context
+def cli(ctx, opt_dir_in_orig, opt_dir_in_new):
+  """Compare quality of super resolution images"""
+
+  import os
+
+  import sys
+  from os.path import join
+  from pathlib import Path
+  from glob import glob
+
+  from random import randint
+  from PIL import Image, ImageOps, ImageFilter
+  from pathlib import Path
+  import cv2 as cv
+
+  from app.settings import types
+  from app.utils import click_utils
+  from app.settings import app_cfg as cfg
+  from app.utils import file_utils, im_utils, logger_utils
+  
+  log = logger_utils.Logger.getLogger()
+
+  fp_ims = glob(join(opt_dir_in_new, '*.jpg'))
+  fp_ims += glob(join(opt_dir_in_new, '*.png'))
+
+  log.info('{}'.format(len(fp_ims)))
+
+  while True:
+    rn = randint(0, len(fp_ims) - 1)
+    fp_im_new = fp_ims[rn]
+    fp_im_orig = fp_im_new.replace(opt_dir_in_new, opt_dir_in_orig)
+    log.info('new: {}'.format(fp_im_new))
+    log.info('orig: {}'.format(fp_im_orig))
+
+    im_new = cv.imread(fp_im_new)
+    im_orig = cv.imread(fp_im_orig)
+
+    # show
+    cv.imshow('new', im_new)
+    cv.imshow('orig', im_orig)
+
+    # handle key io
+    k = cv.waitKey(0) & 0xFF
+    if k == 27 or k == ord('q'):  # ESC
+      # exits the app
+      cv.destroyAllWindows()
+      sys.exit('Exiting because Q or ESC was pressed')
+    elif k == ord(' ') or k == 81 or k == 83:
+      continue
+