Deepface ResourceExhaustedError: не удалось выделить память [Op:AddV2]

Я новичок в глубоком обучении. Я пытаюсь использовать библиотеку deepface на своем локальном компьютере . я использовалpip install deepfaceчтобы установить библиотеку, пробовал на python 3.7.13, 3.8.13 и 3.9.13, которые были созданы с использованием виртуальной среды conda.

Однако при запуске приведенного ниже фрагмента кода я получаю ту же ошибку при работе на моем локальном компьютере. Нужен ли мне графический процессор для запуска библиотеки? Если да, то как мне его настроить? Потому что из онлайн-руководств/статей ни в одном из них не упоминается необходимость установки/настройки графического процессора.

У меня есть GeForce MX450 на моем локальном компьютере.

код

      import cv2
from deepface import DeepFace
import numpy as np

def analyse_face():
    imagepath = "happy_face_woman.png"
    image = cv2.imread(imagepath)
    face_analysis = DeepFace.analyze(image)
    print(face_analysis)

print(analyse_face())

Ошибка:

      ResourceExhaustedError                    Traceback (most recent call last)
~\AppData\Local\Temp\ipykernel_14196\3829791526.py in <module>
     12     print(face_analysis)
     13 
---> 14 analyse_face()

~\AppData\Local\Temp\ipykernel_14196\3829791526.py in analyse_face()
      9     imagepath = "happy_face_woman.png"
     10     image = cv2.imread(imagepath)
---> 11     face_analysis = DeepFace.analyze(image)
     12     print(face_analysis)
     13 

c:\Users\user_name\anaconda3\envs\deepFacepy37\lib\site-packages\deepface\DeepFace.py in analyze(img_path, actions, models, enforce_detection, detector_backend, prog_bar)
    352 
    353         if 'age' in actions and 'age' not in built_models:
--> 354                 models['age'] = build_model('Age')
    355 
    356         if 'gender' in actions and 'gender' not in built_models:

c:\Users\user_name\anaconda3\envs\deepFacepy37\lib\site-packages\deepface\DeepFace.py in build_model(model_name)
     61                 model = models.get(model_name)
     62                 if model:
---> 63                         model = model()
...
-> 1922         seed=self.make_legacy_seed())
   1923 
   1924   def truncated_normal(self, shape, mean=0., stddev=1., dtype=None):

ResourceExhaustedError: failed to allocate memory [Op:AddV2]

Другой вывод ошибок

      ResourceExhaustedError                    Traceback (most recent call last)
~\AppData\Local\Temp\ipykernel_14196\3829791526.py in <module>
     12     print(face_analysis)
     13 
---> 14 analyse_face()

~\AppData\Local\Temp\ipykernel_14196\3829791526.py in analyse_face()
      9     imagepath = "happy_face_woman.png"
     10     image = cv2.imread(imagepath)
---> 11     face_analysis = DeepFace.analyze(image)
     12     print(face_analysis)
     13 

c:\Users\user_name\anaconda3\envs\deepFacepy37\lib\site-packages\deepface\DeepFace.py in analyze(img_path, actions, models, enforce_detection, detector_backend, prog_bar)
    352 
    353         if 'age' in actions and 'age' not in built_models:
--> 354                 models['age'] = build_model('Age')
    355 
    356         if 'gender' in actions and 'gender' not in built_models:

c:\Users\user_name\anaconda3\envs\deepFacepy37\lib\site-packages\deepface\DeepFace.py in build_model(model_name)
     61                 model = models.get(model_name)
     62                 if model:
---> 63                         model = model()
...
-> 1922         seed=self.make_legacy_seed())
   1923 
   1924   def truncated_normal(self, shape, mean=0., stddev=1., dtype=None):

ResourceExhaustedError: OOM when allocating tensor with shape[7,7,512,4096] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc [Op:RandomUniform]

Дополнительная информация Я выполнил команду, чтобы проверить использование моего графического процессора, и подробности следующие:

      !nvidia-smi

0 ответов

Другие вопросы по тегам