1adrianb / 1adrianb/face-alignment
Support for Batch Inference
- 主要言語
- 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