search

black0017 / MedicalZooPytorch

A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation
MIT License
1.72k stars 299 forks source link

pretrained models for COVID-19 CT segmentation #2

Closed JunMa11 closed 4 years ago

JunMa11 commented 4 years ago

Hi @black0017 ,

Thanks for your great work.

I notice that you are also working on pre-trained models on public datasets.

Here is a benchmark for lung and infection segmentation in COVID-19 CT scans.

Are you interested in including this COVID-19 segmentation dataset for pre-trained models?

Best regards, Jun

black0017 commented 4 years ago

Hello and thanks for the comment.

We uploaded a dataloader for covid segmentation dataset (taks1 in your git) and a test file to run with most of our architectures. Feel free to share a link in your coniv benchmark segm repo.

We will provide update on this when we start to realese pre-trained models. It will take some time, since we have a lot of experiments!!!

Before we want to do a massive cleanup on the repo and clarify the API so that people can use it smoothly.

Thanks.