deepinsight / deepinsight/insightface
Error: cudaMalloc failed: out of memory
Open
- Dominant language
- Python
- Stars
- 29.7k
- Forks
- 6.1k
- PR merge metrics
- No merged PRs in 30d
Description
Hello:
I use this code to detect bounding box in multiple images with GPU:
```
model = insightface.app.FaceAnalysis()
ctx_id = 0
model.prepare(ctx_id = ctx_id, nms=0.8)
faces = model.get(img)
for idx, face in enumerate(faces):
box=face.bbox.astype(np.int).flatten()
list_box.append(box)
```
When I process 500 images I get an error:
cudaMalloc failed: out of memory
How can I fix the problem?
Regards,
Roberto
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.