search

MouseLand / cellpose

a generalist algorithm for cellular segmentation with human-in-the-loop capabilities
https://www.cellpose.org/
BSD 3-Clause "New" or "Revised" License
1.38k stars 393 forks source link

Training denoise model #984

Closed walkeraw2002 closed 1 month ago

walkeraw2002 commented 3 months ago

Hello,

I was wondering if it is possible to train the denoise, deblur, and upsample models on our own images for fine-tuning. I see there is documentation for training the segmentation model, but I haven't been able to find information on training the image enhancement models.

Thank you very much!

carsen-stringer commented 1 month ago

we have added the information here: https://cellpose.readthedocs.io/en/latest/restore.html#training-your-own-models

you can use --pretrained_model as an input in the CLI to start from one of our models (or in the API by specifying the model type when creating the DenoiseModel). we have not tested fine-tuning though, so I'm not sure how well it will work, it would be great if you could let us know how it goes