FZJ-INM1-BDA / celldetection

Scalable Instance Segmentation using PyTorch & PyTorch Lightning.
https://docs.celldetection.org
Apache License 2.0
124 stars 21 forks source link

Trained weights? #13

Open fipeop opened 1 year ago

fipeop commented 1 year ago

Hi,

Nice model! Would it be possible to share the trained weights here: ginoro_CpnResNeXt101UNet-fbe875f1a3e5ce2c?

Thank you!

ericup commented 1 year ago

Thanks! Sure, you can either download it automatically via Python

import celldetection as cd

model_name = 'ginoro_CpnResNeXt101UNet-fbe875f1a3e5ce2c'

model = cd.fetch_model(model_name, check_hash=True)  # either plain torch
lit_model = cd.models.LitCpn(model_name)  # or lightning

or download it manually here. I also just added the download link to the main readme.