search

Xiaoqi-Zhao-DLUT / MSNet-M2SNet

(MICCAI 2021) Automatic Polyp Segmentation via Multi-scale Subtraction Network
106 stars 14 forks source link

pretrained models #1

Closed charlie2w closed 2 years ago

charlie2w commented 2 years ago

Hi, I couldn't find the pre-trained models, "res2net50_v1b_26w_4s-3cf99910.pth", used in this code. Wish for your sincerely reply!

charlie2w commented 2 years ago

Hi, I also have a question when I use the train.py

Traceback (most recent call last): File "D:/MSNet-main/train.py", line 85, in train(dataset_medical, MSNet, LossNet) File "D:/MSNet-main/train.py", line 38, in train net = Network(cfg) TypeError: init() takes 1 positional argument but 2 were given

Xiaoqi-Zhao-DLUT commented 2 years ago

@charlie2w We upload the link of Res2Net weights in Readme and update the train.py. You can try it again.

charlie2w commented 2 years ago

@charlie2w We upload the link of Res2Net weights in Readme and update the train.py. You can try it again.

Thanks for your reply. Best wishes!