Torchcam:类激活资源管理器
利用PyTorch中特定于类的卷积层激活的简单方法。
目录
入门
先决条件
Python 3.6(或更新版本)
安装
您可以使用安装软件包,如下所示:
pip install torchcam
或使用 :
conda install -c frgfm torchcam
用法
Torchcam的建立既适合希望更好地了解其CNN模型的用户,也供研究人员使用流行的方法享受强大的实施基础。 这是一个简短的片段,说明了其用法:
import torch
from torchcam . cams import SmoothGradCAMpp
from torchvision . models import resnet18
img_tensor = torch . rand (( 1 , 3 , 224 , 224 ))
model = resnet18
1