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