torchextractor:torchxtractor使特征提取变得简单 源码

割炬:PyTorch中间特征提取 介绍 仅仅因为forward功能未返回用户期望的内容,许多模型定义就被无情地复制粘贴了。 使用torchextractor您可以轻松捕获中间特征图,并在其上构建任何您喜欢的东西。 安装 pip install git+https://github.com/antoinebrl/torchextractor.git 用法 import torch import torchvision import torchextractor as tx model = torchvision . models . resnet18 ( pretrained = True ) model = tx . Extractor ( model , [ "layer1" , "layer2" , "layer3" , "layer4" ]) dummy_input = torch