ChaoningZhang / MobileSAM

This is the official code for MobileSAM project that makes SAM lightweight for mobile applications and beyond!
Apache License 2.0
4.86k stars 505 forks source link

Request-Code for counting objects using text prompt in an image and domain specific finetuning script #77

Open AT38875 opened 1 year ago

AT38875 commented 1 year ago

Hi @ksugar @dhkim2810 @qiaoyu1002 @ChaoningZhang Can you also incorporate the code for counting objects in an image based on prompts. Also it will be great if you could publish the fine-tuning script for custom objects. Here is the example for counting objects. https://github.com/ylqi/Count-Anything/ It is using SAM model which takes roughly 7GB GPU memory for inference and around 15-20 minutes. Also throwing error for custom images.Can you incorporate those changes in mobile SAM for object counter as well. Thanks