TIO-IKIM/CellViT

Train a data set with only one tissue type

Transformer-man opened this issue · 3 comments

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?

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

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

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).