deepinsight / deepinsight/insightface

how use gpu to infer and generate mask readerer data

Open
#1,743 3 comments 0 reactions 0 assignees View on GitHub
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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.