deepinsight / deepinsight/insightface
how use gpu to infer and generate mask readerer data
- Dominant language
- Python
- Stars
- 29.7k
- Forks
- 6.1k
- PR merge metrics
- No merged PRs in 30d
Description
In ./recongnition/_tools/mask_render.py, II executed `CUDA_VISIBLE_DEVICES=0 python mask_renderer.py t`he utilization rate of GPU is 0, but the CPU is used to complete the data generation. How to use gpu to realize the generation of face mask data
In the readme document, the following method is to use gpu. I tried it, but to no avail。
`tool.prepare(ctx_id=0, det_size=(128,128)) #use gpu`
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by inspecting ./recongnition/_tools/mask_render.py and the README example using tool.prepare(ctx_id=0, det_size=(128,128)). Reproduce the reported CUDA_VISIBLE_DEVICES=0 command, accounting for the mask_render.py versus mask_renderer.py naming shown in the issue, and verify that face-mask data generation uses the GPU rather than the CPU.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- computer-vision
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100