deepinsight / deepinsight/insightface

Error: cudaMalloc failed: out of memory

Open
#1,158 4 comments 2 reactions 0 assignees View on GitHub
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.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.