1adrianb / 1adrianb/face-alignment

GPU memory consumption

Ouverte
#228 2 commentaires 0 réactions 0 personnes assignées Voir sur GitHub
question
Langage dominant
Python
Étoiles
7.5k
Forks
1.4k
Métriques de merge des PR
Aucune PR mergée en 30 j

Description

Hi to everyone, I'm testing your library. I'm really interested to estimate the facial landmarks. I have a computer with a GPU, and I'm using the **blazeface** detector, which is the faster as it comments in the documentation. **After several tests, the memory is increasing until ~6.5 GB of GPU when I want to extract 2D landmarks and ~8.5 to extract 3D landmarks**. I've noticed as well the **size network** used is always "Large" and there is some other 3 types uncommented, so I'm wondering about the relation of it with GPU performance and consuming because I need to decrease in some way the amount of memory.

Thanks in advance.

Guide de contribution

Aucun guide de contribution indexé pour ce dépôt

Piste de recherche

The issue mentions using the blazeface detector and the 'Large' network size. Start by examining the model loading and inference code in the face-alignment library, particularly around GPU memory management. Look for configuration options for network size and detector settings. Profile memory usage during landmark extraction to identify leaks or inefficient allocations. The goal is to understand why memory grows to 6.5-8.5 GB and if switching network sizes affects this.

Rédigé par le modèle d'indexation à partir du texte de l'issue.

Évaluation

Stack technique
python, pytorch
Domaine
performance
Type d'issue
Bug
Difficulté
4/5
Temps estimé
3-5 jours
Activité
À l'abandon
Clarté
Plutôt claire
Accessibilité débutants
35/100

Recevez les nouvelles issues par e-mail

Un résumé court des issues GitHub adaptées aux débutants.