NVIDIA-AI-IOT / nanosam

A distilled Segment Anything (SAM) model capable of running real-time with NVIDIA TensorRT

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Does it support multi-target real-time detection and segmentation?

HaoqianSong opened this issue · comments

Hi,
Thanks for your great work!
I have successfully run your example 4. The camera can collect and segment a single person in real time, but I do not know whether it supports real-time detection and segmentation of multiple people.

commented

Hi @yangguanghaiann ,

Thanks for the feedback!

Currently example 4 assumes one person in the scene, but I'll try to update it soon to handle multiple detections.

While the image encoder only needs to run once per detection, the mask decoder needs to run for each mask you want to generate. So depending on the number of masks generated, the framerate may vary.

That said, it may still run real-time for a certain number of people.

Hope this helps.

John