search

yxlijun / DSFD.pytorch

DSFD implement with pytorch
167 stars 58 forks source link

resnet50 resnet101 pretrained model #23

Open XDUSPONGE opened 5 years ago

XDUSPONGE commented 5 years ago

hello may I have your resnet50 resnet101 pretrained model thanks very much

ymmshi commented 5 years ago

download pretrained model https://download.pytorch.org/models/resnet152-b121ed2d.pth or https://download.pytorch.org/models/resnet50-19c8e357.pth and then modify 121 line in train.py as follows: pretrained_dict = base_weights model_dict = net.resnet.state_dict() pretrained_dict = {k:v for k,v in pretrained_dict.items() if k in model_dict} model_dict.update(pretrained_dict) print(model_dict) net.resnet.load_state_dict(model_dict)

pacchomp commented 1 year ago

download pretrained model https://download.pytorch.org/models/resnet152-b121ed2d.pth or https://download.pytorch.org/models/resnet50-19c8e357.pth and then modify 121 line in train.py as follows: pretrained_dict = base_weights model_dict = net.resnet.state_dict() pretrained_dict = {k:v for k,v in pretrained_dict.items() if k in model_dict} model_dict.update(pretrained_dict) print(model_dict) net.resnet.load_state_dict(model_dict)

你好，请问有没有res152训练完在wider face上的结果