HuCaoFighting / Swin-Unet

[ECCVW 2022] The codes for the work "Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentation"
1.69k stars 306 forks source link

Pretraining of the model #97

Open hyukiggle opened 1 year ago

hyukiggle commented 1 year ago

I was trying to reproducing the equivalent result from the original paper. Though it is natural to pretrain encoder of model on ImageNet, the decoder is task-specifically trained. As of I know, since ImageNet pretraining is classification, it is difficult to train decoder. I'd like to know how the pretrained decoder weight is initialized or the method of decoder's pretraining on ImageNet. Thank you

manupillai308 commented 5 months ago

@HuCaoFighting