Revert "Fix to an UnboundLocalError due to rename. (deepfakes#318)"

This reverts commit 2e2dc84.
kvrooman · Mar 28, 2018 · a96e062 · a96e062
1 parent e7db8ee
commit a96e062
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/FaceLandmarksExtractor/FaceLandmarksExtractor.py b/lib/FaceLandmarksExtractor/FaceLandmarksExtractor.py
@@ -173,7 +173,7 @@ def extract(input_image_bgr, detector, verbose, all_faces=True, input_is_predete
             center[1] -= (bottom - top) * 0.12
             scale = (right - left + bottom - top) / 195.0
 
-            image = crop(input_image_bgr, center, scale).transpose ( (2,0,1) ).astype(np.float32) / 255.0
+            image = crop(input_image, center, scale).transpose ( (2,0,1) ).astype(np.float32) / 255.0
             image = np.expand_dims(image, 0)
 
             pts_img = get_pts_from_predict ( keras_model.predict (image)[-1][0], center, scale)