TIO-IKIM / CellViT

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

Train a data set with only one tissue type #21

Closed Transformer-man closed 10 months ago

Transformer-man commented 10 months ago

Hello, my data set has only one tissue type. How do I modify the training configuration? Or do you provide training configuration here? Do we need to close the issue type branch directly, or do we need to manually modify the network?

FabianHoerst commented 10 months ago

If you just have on tissue type, just delete the tissue-branch or provide always the same label in your dataset

Transformer-man commented 10 months ago

image Can I directly set the weight of the organization to 0 without calculating its loss?

FabianHoerst commented 10 months ago

It is possible from machine learning theory. You could also remove the entire tissue_types loss part in the config, but in both cases, the scripts will fail. You need to remove everything associated with the tissue_type calculation, e.g., the tissue-related metrics. But it should be really straightforward. Write your own data loader and fix every error that occurs related to tissue types (e.g., missing dictionary entries in prediction and ground truth).