Bugfix: The propagator fails to process images with alpha channels

Author: suquark

photo_smooth.py

@@ -18,8 +18,8 @@ def __init__(self, beta=0.9999):
     self.beta = beta
 
   def process(self, initImg, contentImg):
-    content = scipy.misc.imread(contentImg)
-    B = scipy.misc.imread(initImg).astype(np.float64)/255
+    content = scipy.misc.imread(contentImg, mode='RGB')
+    B = scipy.misc.imread(initImg, mode='RGB').astype(np.float64)/255
     h1,w1,k = B.shape
     h = h1 - 4
     w = w1 - 4