TIO-IKIM / CellViT

CellViT: Vision Transformers for Precise Cell Segmentation and Classification
https://doi.org/10.1016/j.media.2024.103143
Other
217 stars 33 forks source link

How about the segmentation performance on other benchmarks like CoNSeP, CPM-17? #24

Closed windygooo closed 12 months ago

windygooo commented 12 months ago

And would the pretrained weights on each PanNuke fold be released?

FabianHoerst commented 12 months ago

Hello, thanks for your inquiry!

We have not trained the networks on other datasets, but validated the results on Monuseg and CoNSeP. In our opinion, there is no other dataset that is comparable to PanNuke. See the Table attached, which is from the Paper: Lizard: A Large-Scale Dataset for Colonic Nuclear Instance Segmentation and Classification.

Bildschirm­foto 2023-10-09 um 08 02 02

Training on Lizard would be possible, but the test dataset is not publicly available and the dataset is just from one tissue type. We therefore decided to use the other datasets (Monuseg, CoNSeP) to validate our PanNuke trained networks. Sharing all pretrained network weights is not possible, as each checkpoint encompasses multiple GB and the whole result directory with each checkpoint for all three folds needs more than 1TB of storage. However, you can retrain the networks with our provided training scripts by yourself if this is necessary.

windygooo commented 12 months ago

Thanks for your reply! Will the training config files for the PanNuke dataset be released?

windygooo commented 12 months ago

Thanks for your reply! Will the training config files for the PanNuke dataset be released?

I have seen it.