1adrianb / 1adrianb/face-alignment

GPU memory consumption

未关闭
#228 2 条评论 0 个 reaction 已指派 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 摘要。