1adrianb / 1adrianb/face-alignment

Support for Batch Inference

オープン
#115 コメント 5 件 リアクション 0 件 担当者 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 を短くまとめたダイジェスト。