1adrianb / 1adrianb/face-alignment

GPU memory consumption

Aperta
#228 2 commenti 0 reazioni 0 assegnatari Vedi su GitHub
question
Lingua principale
Python
Stelle
7.5k
Fork
1.4k
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

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.

Guida per i contributori

Nessuna guida per i contributori indicizzata per questo repository

Direzione di ricerca

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.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Valutazione

Stack tecnologico
python, pytorch
Ambito
performance
Tipo di issue
Bug
Difficoltà
4/5
Tempo stimato
3-5 giorni
Stato di attività
Ferma
Chiarezza
Abbastanza chiara
Idoneità per principianti
35/100

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.