第一步:提取分类前网络的中间层的特征
问题:如何知道你需要的中间层的特征层的名称呢?
答案:
1、直接看网络的源码
2、prin他(网络模型)
连接:这个讲的很好!
换backbone的一个难点在于:
如何去获取分类网络中间某一个特征层的一个输出,在这个特征层输出的基础上去构建Faster Rcnn的backbone。
pytorch官方给的方法去构建新的backbone—目标是:将faster Rcnn的backbone用EfficientNetB0、VGG16、resnet50替换
首先将torch升级到1.10的版本
torchvision:与torch1.10对应的那个版本。