xinntao / facexlib

FaceXlib aims at providing ready-to-use face-related functions based on current STOA open-source methods.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Loading bunch of models

aycaecemgul opened this issue · comments

commented

Hello, in face_restoration_helper.py i see that you are using different models: face detection, landmark detection and a segmentation model. This causes cuda out of memory errors in even GPUs with high memory. Why did you use different models and not just use one or two that can give you the same outputs?