1adrianb / 1adrianb/face-alignment

GPU memory consumption

オープン
#228 コメント 2 件 リアクション 0 件 担当者 0 名 GitHub で見る
question
主要言語
Python
スター
7.5k
フォーク
1.4k
PR マージ指標
30日以内にマージされた PR はありません

説明

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.

コントリビューションガイド

このリポジトリのコントリビューションガイドは索引されていません

調査の方向性

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.

索引モデルが issue の本文から書いたものです。

評価

技術スタック
python, pytorch
領域
performance
issue の種類
バグ
難易度
4/5
見積もり時間
3〜5日
活発さ
停滞
明瞭さ
おおむね明確
初心者へのやさしさ
35/100

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。