Skip to content

wenjing-gg/Gradcam-3D

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Gradcam-3D

Table of Contents

Introduction

This is a code for generating 3D gradcam, I hope it will be useful for you

Preparation

If you want to use this code, first you need to change a few pieces of code in cam.py

using your own dataset

from dataset import Dataset

import your model

from models.resnet_50 import resnet50_3d

Change to the target_layer of the model you want to generateChange to the target_layer of the model you want to generate

target_layers = [model.layer1]

Example

The result is as follows: Result

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%