代码之家 › 专栏 › 技术社区 › Vipin Chaudhary

如何从pytorch中预训练的CNN的特定层输出

resnet pytorch deep-learning

Vipin Chaudhary · 技术社区 · 8 年前

我已经在imagenet数据集上对CNN(RESNET18)进行了预训练,现在我想要的是从特定层获得输入图像的输出,

例如

我的输入图像是 FloatTensor(3, 224, 336) 我发了一批 size = 10 在我的resnet模型中,现在我想要的是模型返回的输出。第4层,

现在我尝试的是 out = model.layer4(Variable(input)) 但它给了我输入维度不匹配的错误( ),这就是返回的确切错误

运行时错误:需要输入维度4和输入。尺寸[1]==64,但在/Users/soumith/miniconda2/conda bld/pytorch_1501999754274/work/torch/lib/THNN/generic/SpatialConvolutionMM处获得形状输入:[10 x 3 x 224 x 336]。c: 47

所以我很困惑,现在该如何获得第4层的输出

1 回复 | 直到 8 年前

AmirHossein 8 年前

您必须创建一个包含从开始到所需块的所有层的模块:

resnet = torchvision.models.resnet18(pretrained=True)
f = torch.nn.Sequential(*list(resnet.children())[:6])
features = f(imgs)

推荐文章

Saffy · 如何在IterableDataset上应用最小最大缩放?

8 月前

sanjeev mk · 通过索引从Pytorch或Numpy 2D数组中快速删除多行的方法

1 年前

Anonymous · 如何为零维火炬张量赋值?

1 年前

JohnnyWang97 · getattr引起的有趣错误

1 年前

Kamugg · 在PyTorch中使用不同分辨率图像训练DeepLabV3的最佳实践

1 年前

Stocavista · 无法在python中将float 64转换为float 32

1 年前

efwefwefwefwefw wefwefwefwef · 如何在PyTorch Conv1d层中仅在一侧应用填充?

1 年前

Okhr · 运行时错误:CUDA错误:在带有GTX 1660 Super的Debian 12虚拟机上不支持此操作

1 年前

Fatemeh · 如何从使用nn训练和保存的模型加载检查点。DataParallel到不使用nn的模型上。DataParallel?

1 年前

Twenkid · 将GPT2 h5型号转换为割炬,以转换为ggml形状不匹配

1 年前