adapt PY2 PY3 (#326)

demo/slimfacenet/dataloader/casia.py

@@ -13,7 +13,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import numpy as np
-import imageio
+import six
+if six.PY2:
+    import scipy.misc as imgreader
+else:
+    import imageio as imgreader
 import os
 import paddle
 from paddle import fluid
@@ -54,7 +58,7 @@ class CASIA_Face(object):
             target = self.label_list[index]
 
             try:
-                img = imageio.imread(img_path)
+                img = imgreader.imread(img_path)
             except:
                 continue
 

demo/slimfacenet/dataloader/lfw.py

@@ -13,7 +13,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import numpy as np
-import imageio
+import six
+if six.PY2:
+    import scipy.misc as imgreader
+else:
+    import imageio as imgreader
 import paddle
 from paddle import fluid
 
@@ -32,10 +36,10 @@ class LFW(object):
                 return
             index = self.shuffle_idx.pop(0)
 
-            imgl = imageio.imread(self.imgl_list[index])
+            imgl = imgreader.imread(self.imgl_list[index])
             if len(imgl.shape) == 2:
                 imgl = np.stack([imgl] * 3, 2)
-            imgr = imageio.imread(self.imgr_list[index])
+            imgr = imgreader.imread(self.imgr_list[index])
             if len(imgr.shape) == 2:
                 imgr = np.stack([imgr] * 3, 2)