1adrianb / 1adrianb/face-alignment
Support for Batch Inference
- Ngôn ngữ chính
- Python
- Star
- 7.5k
- Fork
- 1.4k
- Chỉ số merge pull request
- Không có pull request nào được merge trong 30 ngày
Mô tả
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.
Hướng dẫn đóng góp
Chưa lập chỉ mục được hướng dẫn đóng góp cho kho mã nguồn này
Hướng nghiên cứu
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.
Do mô hình lập chỉ mục viết ra từ nội dung của issue.
Đánh giá
- Công nghệ
- python, pytorch
- Lĩnh vực
- ai, machine-learning, performance
- Loại issue
- Tính năng
- Độ khó
- 4/5
- Thời gian dự kiến
- 3-5 ngày
- Mức độ hoạt động
- Đình trệ
- Độ rõ ràng
- Khá rõ ràng
- Mức phù hợp với người mới
- 45/100