search

wwyi1828 / CluSiam

Improving Representation Learning for Histopathologic Images with Cluster Constraints
MIT License
16 stars 1 forks source link

How to inference on cluster task? #7

Closed lsnls closed 2 months ago

lsnls commented 3 months ago

Hello, thank you for your great work!

How do I use the weights of the trained model to cluster images? I only found the code for model training in the repository.

I am looking forward to your reply. Thank you very much!

wwyi1828 commented 3 months ago

Hi lsnls,

Thanks for your interest in our work! To use the trained model weights for clustering images, you can directly take the argmax of the output from model.cluster_assigner. the clustering task in our work is designed as an auxiliary pretext task to help the model learn useful representations. I did not initially expect the learned clusters can align well with the tumor/non-tumor in the C16.