xinntao / facexlib

FaceXlib aims at providing ready-to-use face-related functions based on current STOA open-source methods.
MIT License
810 stars 143 forks source link

Loading bunch of models #20

Open aycaecemgul opened 2 years ago

aycaecemgul commented 2 years ago

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?