Faster RCNN如何换特征层

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

Faster RCNN如何换特征层_第1张图片

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

答案:
1、直接看网络的源码
2、prin他(网络模型)

连接:这个讲的很好!

换backbone的一个难点在于:

如何去获取分类网络中间某一个特征层的一个输出,在这个特征层输出的基础上去构建Faster Rcnn的backbone。

pytorch官方给的方法去构建新的backbone—目标是:将faster Rcnn的backbone用EfficientNetB0、VGG16、resnet50替换

首先将torch升级到1.10的版本
torchvision:与torch1.10对应的那个版本。

Faster RCNN如何换特征层_第2张图片

你可能感兴趣的:(代码文件,计算机视觉,深度学习)