1adrianb / 1adrianb/face-alignment

Memory not cleansed on CPU

Offen
#185 1 Kommentar 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
Vorherrschende Sprache
Python
Sterne
7.5k
Forks
1.4k
PR-Merge-Kennzahlen
Keine gemergten PRs in 30 T.

Beschreibung

When I use
`face_alignment.FaceAlignment(face_alignment.LandmarksType._3D, device='cuda')`
or
`face_alignment.FaceAlignment(face_alignment.LandmarksType._3D, face_detector='dlib')`
everything's fine.
However, when I use
`face_alignment.FaceAlignment(face_alignment.LandmarksType._3D, device='cpu')`
some garbage accumulates in memory and with every use the process that runs `get_landmarks_from_image` starts consuming more and more memory.

Beitragsleitfaden

Für dieses Repository ist kein Beitragsleitfaden indexiert

Rechercherichtung

The issue is a memory leak when using the CPU device. Start by examining the FaceAlignment class initialization and the get_landmarks_from_image method in the face_alignment module. Look for any unmanaged resources, caching, or state that accumulates on CPU but not on CUDA. Run a simple script that repeatedly calls get_landmarks_from_image on CPU and monitor memory usage with tools like memory_profiler. Check if the leak is in the underlying PyTorch operations or the face detection backend.

Vom Indexierungsmodell aus dem Issue-Text verfasst.

Bewertung

Tech-Stack
python, pytorch
Bereich
performance
Issue-Typ
Bug
Schwierigkeit
3/5
Geschätzter Aufwand
1-2 Tage
Aktivitätsstatus
Veraltet
Klarheit
Klar beschrieben
Anfängerfreundlichkeit
45/100

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.