MontaEllis / Pytorch-Medical-Segmentation

This repository is an unoffical PyTorch implementation of Medical segmentation in 2D and 3D.
MIT License
847 stars 197 forks source link

Sharing models through Hugging Face Hub #42

Open omarespejel opened 2 years ago

omarespejel commented 2 years ago

Hi MontaEllis team!

The Medical Segmentation project is fantastic. I see you host and share models with your server. Would you be interested in sharing your models in the Hugging Face Hub?

This integration would allow you to freely download/upload models and make your work more accessible and visible to the rest of the ML community. We can help you set up a MontaEllis organization (examples, Facebook AI y Stanford NLP).

Creating the repos and adding new models should be a relatively straightforward process. This is a step-by-step guide explaining the process in case you're interested. Please let us know if you would be interested and if you have any questions.

Some of the benefits of sharing your models through the Hub would be:

Additionally, we have a library to access repositories programmatically (both downloading pre-trained models and pushing, with many nice things such as filtering, caching, etc.). If we want to try out this integration, I would suggest you add one or two models manually and then use the huggingface_hub library to implement downloading those models programmatically from pytorch-medical-segmentation. To read more about it, you might want to check our documentation to read more about it.

Relevant references:

Happy to hear your thoughts,

Omar and the Hugging Face team (cc @osanseviero @abidlabs )

omarespejel commented 2 years ago

Hi @MontaEllis!

We are very interested in giving even more visibility and support to the deep learning for the healthcare community and your segmentation models are key .

We have new collaboration features in the Hub: pull requests and discussions (link to announcement).

This would allow medical segmentation users to submit, edit, or discuss models within a possible MontaEllis organization in the Hub.

Also, if you want, we can allow image segmentation inference directly from the browser, for example with DETR, for MontaEllis models.

Please let me know if we can help further. Kind regards, Omar