Unable to process images which are received from socketio image buffer about insightface-rest HOT 12 CLOSED

By reading few blogs I found that this error has something to do with cuda context and this can be solved by

saving the CUDA context.
Issue here is CUDA context refreshed and mixed up with other applications.

Can you help me out now?

from insightface-rest.

Could you please provide more info on how image is sent and received?

from insightface-rest.

I am connected to a server !

Server is sending me an image in byte array format.

Then I am converting that byte array into numpy array and then processing it further.

I am only receiving image , not sending!

from insightface-rest.

It's hard to tell which reason might cause this exception without additional info, but I think it's mostly connected to the way you read the image before feeding it for inference.
I have tested this code with reading images from rabbitmq query, and from fastapi endpoint, which is default behavior of this project, and haven't got such exceptions.
If you could provide minimal reproducible example I could try checking it.

from insightface-rest.

Could you also provide minimum server example? For testing full cycle.

EDIT: Have you tried to save image decoded by opencv to disk with cv.imwrite?

from insightface-rest.

Yes , image is perfect the way it should be !
There is no issues with the input image that's for sure ! I have cross checked with storing that image in local storage and also checked it , it is how it should be !!

from insightface-rest.

Found the solution!!
Closing it

from insightface-rest.

Great news, @bhargavravat! Could you please post your solution? I'll apply it to master branch if it doesn't conflict with FastAPI.

from insightface-rest.

I have modified trt_loader.py (/app/modules/model_zoo/exec_backends/trt_loader.py)

Attaching it as .txt format

trt_loader.txt

from insightface-rest.

I have tested your solution with /src/converters/pipeline_tester.py, but it fails with following error:

PyCUDA ERROR: The context stack was not empty upon module cleanup.

I was able to fix it by replacing line 84:

self.cuda_ctx = cuda_ctx

with:

device = cuda.Device(0)  # enter your gpu id here
ctx = device.make_context()
self.cuda_ctx = ctx

This worked, but caused excessive GPU RAM usage, about 1GB

from insightface-rest.

hello @SthPhoenix how to create 1 engine (ex face detection) serve for more than 2 camera using threading?
i tried, but i got confused outputs between these thread.

from insightface-rest.

Hi @ThiagoMateo ! As I have said before, this use case is not tested, for now you can use provided rest API in your application.
In future versions I'm planning to add full support of Triton Inference Server, which will give you ability to serve single model for multiple threads.

from insightface-rest.