routine refactors

Author: QiuJueqin

README.md

@@ -34,9 +34,12 @@ Clone this repo and run
 python demo.py
 ```
 
-The ISP outputs will be saved to `./raw` directory.
+The ISP outputs will be saved to `./output` directory.
+
+The only required package for pipeline execution is `numpy`. 
+
+`opencv-python` and `scikit-image` are required for data IO. 
 
-The only required package for pipeline execution is `Numpy`. `opencv-python` and `scikit-image` are required for data IO. 
 
 # Algorithms
 

demo.py

@@ -1,3 +1,6 @@
+import os
+import os.path as op
+
 import cv2
 import numpy as np
 import skimage.io
@@ -6,6 +9,10 @@
 from utils.yacs import Config
 
 
+output_dir = './output'
+os.makedirs(output_dir, exist_ok=True)
+
+
 def demo_test_raw():
     print('Processing test raw started')
 
@@ -18,7 +25,7 @@ def demo_test_raw():
 
     data, _ = pipeline.execute(bayer)
 
-    output_path = raw_path.replace('.RAW', '.png')
+    output_path = op.join(output_dir, 'test.png')
     output = cv2.cvtColor(data['output'], cv2.COLOR_RGB2BGR)
     cv2.imwrite(output_path, output)
 
@@ -34,7 +41,7 @@ def demo_nikon_d3x():
 
     data, _ = pipeline.execute(bayer)
 
-    output_path = pgm_path.replace('.pgm', '.png')
+    output_path = op.join(output_dir, 'color_checker.png')
     output = cv2.cvtColor(data['output'], cv2.COLOR_RGB2BGR)
     cv2.imwrite(output_path, output)