search

bubbliiiing / faster-rcnn-pytorch

这是一个faster-rcnn的pytorch实现的库，可以利用voc数据集格式的数据进行训练。
MIT License
1.55k stars 352 forks source link

feature和classifier #73

Open duanhan123 opened 3 years ago

duanhan123 commented 3 years ago

大佬 class Resnet50里 features = list([model.conv1, model.bn1, model.relu, model.maxpool, model.layer1, model.layer2, model.layer3]) classifier = list([model.layer4, model.avgpool]) feature是不完整的resnet50，有什么讲究吗？还有classifier为什么要用layer4，盼解惑

bubbliiiing commented 3 years ago

emm我看别人就是这么设计的……

swjtulinxi commented 3 years ago

大佬，你这个训练的时候怎么导入backbone的预训练权重啊，我看你这个导入的是你训练好的整个模型的权重，我们一般第一次训练不是要导入一个backbone的预训练吗

bubbliiiing commented 3 years ago

在构建主干的时候，完成构建时载入，你看看torchvision