1adrianb / 1adrianb/face-alignment

Support for Batch Inference

未关闭
#115 5 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看
主要语言
Python
星标
7.5k
派生
1.4k
PR 合并指标
30 天内没有已合并 PR

描述

Hi and Thanks for the great package.

I wanted to ask if there is any support for inferring Landmarks of a batch of images? Each `fa.get_landmarks(image)` takes a really long time compared to something like DLib's landmark detector even on a GPU and I was hoping I could use batch processing to process a batch of images at once and remedy this issue a bit.

Thanks again.

贡献指南

这个仓库没有索引到贡献指南

调研方向

Look at the inference code in face_alignment/api.py and the underlying model forward passes. The issue is about modifying the pipeline to accept and process a batch of images instead of one at a time. Check how images are currently loaded and passed to the model, and see if the model itself supports batched input. 'Done' means the get_landmarks function can accept a list of images and return a list of landmarks efficiently.

由索引模型根据 Issue 内容生成。

评估

技术栈
python, pytorch
领域
ai, machine-learning, performance
Issue 类型
功能
难度
4/5
预计耗时
3-5 天
活跃度
停滞
描述清晰度
基本清楚
新手友好度
45/100

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。