diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py index 565c6e5..c61c31b 100644 --- a/deepface/DeepFace.py +++ b/deepface/DeepFace.py @@ -695,7 +695,8 @@ def represent( # represent if "keras" in str(type(model)): - # new tf versions show progress bar and it is annoying + # model.predict causes memory issue when it is called in a for loop + # embedding = model.predict(img, verbose=0)[0].tolist() embedding = model(img, training=False).numpy()[0].tolist() else: # SFace and Dlib are not keras models and no verbose arguments