How can I write a generator based on this foundation

NVIDIA-AI-IOT / nanosam

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

Apache License 2.0

616 stars 52 forks source link

How can I write a generator based on this foundation #5

Open Xcccck opened 11 months ago

Xcccck commented 11 months ago

Hi, Thanks for this good job!! I have completed the use of Predictor, now I want to implement a Generator just like the code in the mobileSam But i dont know how to do it , it's so hard for me , i need some help Thanks a lot ''' like this , i want to get whole masks by Generator''' def get_mask(srcImg, model): mask_generator = SamAutomaticMaskGenerator(model) masks = mask_generator.generate(srcImg) return masks

hassaniqbal209 commented 7 months ago

Hi, I am interested in the same problem. Were you able to try this?