Merge branch 'master' of https://github.com/Shawn-Shan/fawkes

fawkes/utils.py

@@ -1,3 +1,4 @@
+import errno
 import glob
 import gzip
 import json
@@ -259,9 +260,9 @@ def fix_gpu_memory(mem_fraction=1):
 def preprocess(X, method):
     assert method in {'raw', 'imagenet', 'inception', 'mnist'}
 
-    if method is 'raw':
+    if method == 'raw':
         pass
-    elif method is 'imagenet':
+    elif method == 'imagenet':
         X = imagenet_preprocessing(X)
     else:
         raise Exception('unknown method %s' % method)
@@ -272,9 +273,9 @@ def preprocess(X, method):
 def reverse_preprocess(X, method):
     assert method in {'raw', 'imagenet', 'inception', 'mnist'}
 
-    if method is 'raw':
+    if method == 'raw':
         pass
-    elif method is 'imagenet':
+    elif method == 'imagenet':
         X = imagenet_reverse_preprocessing(X)
     else:
         raise Exception('unknown method %s' % method)