search

xinghaochen / TinySAM

Official PyTorch implementation of "TinySAM: Pushing the Envelope for Efficient Segment Anything Model"
Apache License 2.0
403 stars 23 forks source link

Have weights available #1

Closed merveenoyan closed 11 months ago

merveenoyan commented 11 months ago

Hello 👋 Thanks a lot for open-sourcing the code! I wanted to try TinySAM and in the demo and README there seems to be a weights folder which I couldn't find.

In any case, would you like to have the weights available on Hugging Face Hub and load them from there?

xinghaochen commented 11 months ago

Checkpoint is released in https://github.com/xinghaochen/TinySAM/releases/download/1.0/tinysam.pth

Thanks for the suggestion and we may upload the weights to huggingface later.

merveenoyan commented 11 months ago

Hello @xinghaochen I've uploaded model weights and built a demo: https://huggingface.co/spaces/merve/tinysam https://huggingface.co/merve/tinysam Would you like to add this to README so people can try easily?

xinghaochen commented 11 months ago

Thanks for the awesome demo! I have just added the link to your Hugging Face demo in README.

merveenoyan commented 11 months ago

@xinghaochen do you (or your lab/organization) have a hugging face account? We can move the repositories under your namespace if so.

xinghaochen commented 11 months ago

@xinghaochen do you (or your lab/organization) have a hugging face account? We can move the repositories under your namespace if so.

https://huggingface.co/xinghaochen

merveenoyan commented 11 months ago

@xinghaochen we noticed there's an organization for Huawei Noah's Ark Lab: https://huggingface.co/huawei-noah you can join there and move the models and demo there if it works for you (we moved all of them under your workspace, you can directly go to repository settings and you'll find the option to move it to the organization once you become a member of the org)