openmedlab / USFM

215 stars 16 forks source link

[Pretrained Model] Can I use Classification Pretrained? #24

Open dablro12 opened 3 days ago

dablro12 commented 3 days ago

image

this code is not operated.

How to use classifier model using pretrained weight?

When can I use pre-trained models for classification tasks?

BodongDu commented 1 day ago

I have the same questions with you. The different model in different tasks, one is SegVit and the other is Vit. The pulished weight only fit the Segmentation task. And it's not consistent with the paper. I didn't really understand why the project use two pre-training models for two tasks?