search

anthonyweidai / SvANet

Official implementation of "Exploiting Scale-Variant Attention for Segmenting Small Medical Objects"
MIT License
28 stars 1 forks source link

How can I download the pretrained model? #2

Closed Stupid-wangnz closed 1 month ago

Stupid-wangnz commented 1 month ago

How can I download the pretrained model mentioned in the training script. "--pretrained 2 --freeze_weight 3 --weight_name resnet50-2.pth"

Stupid-wangnz commented 1 month ago

I didn't see any information about which pretrained model the encoder model used in the paper, maybe I overlooked it. May I ask if you can provide more information about the pretrained encoder used.

silentyuchen commented 1 month ago

I'm using wide-resnet-50-2-export-5ae25d50.pth, it works

anthonyweidai commented 1 month ago

We used pretrained model downloaded from torchvision official website. For resnet50-2.pth, you can download it by this link.

anthonyweidai commented 1 month ago

The difference between IMAGENET1K_V1 and IMAGENET1K_V2 versions of the pretrained model lies in different training configurations.