Faster RCNN如何换特征层

第一步:提取分类前网络的中间层的特征

在这里插入图片描述

问题:如何知道你需要的中间层的特征层的名称呢?

答案:
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版权协议,转载请附上原文出处链接和本声明。