diff --git a/requirements.txt b/requirements.txt index b7418c9ac..48014ef44 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ tensorflow==1.7 -scipy +scipy==1.1.0 scikit-learn opencv-python h5py diff --git a/src/align/detect_face.py b/src/align/detect_face.py index 7f98ca7fb..8c0b61d01 100644 --- a/src/align/detect_face.py +++ b/src/align/detect_face.py @@ -82,7 +82,7 @@ def load(self, data_path, session, ignore_missing=False): session: The current TensorFlow session ignore_missing: If true, serialized weights for missing layers are ignored. """ - data_dict = np.load(data_path, encoding='latin1').item() #pylint: disable=no-member + data_dict = np.load(data_path, encoding='latin1', allow_pickle=True).item() #pylint: disable=no-member for op_name in data_dict: with tf.variable_scope(op_name, reuse=True):