bubbliiiing / centernet-pytorch

这是一个centernet-pytorch的源码,可以用于训练自己的模型。

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

你好

wujinxiao opened this issue · comments

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

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

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

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