search

bubbliiiing / centernet-pytorch

这是一个centernet-pytorch的源码，可以用于训练自己的模型。
MIT License
364 stars 99 forks source link

你好 #24

Open wujinxiao opened 2 years ago

wujinxiao commented 2 years ago

你好 我想问一下 我将Resnet换成了SENet 请问如果预训练权重不可使用了 那我该怎么做才能训练

bubbliiiing commented 2 years ago

应该直接有se-resnet那种库吧，都提供权值的

wujinxiao commented 2 years ago

你好 我找到了se-resnet的库： import torch.hub hub_model = torch.hub.load( 'moskomule/senet.pytorch', 'se_resnet50', pretrained=True,) 请问需要把这个加在哪里

bubbliiiing commented 2 years ago

那你就根据现在有的这个主干修改一下呗，载入权值和网络加载的部分不都有吗