torch cam:您的PyTorch模型(CAMGrad CAMGrad CAM ++Smooth Grad CAM ++Score CAMSS CAMIS
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
用户评论