Stephenfang51 / Grad_CAM_Pytorch-1.01

CNN可视化代码,帮助了解建立GradCam过程, 代码中文注解
100 stars 24 forks source link

请问有训练的代码吗? #1

Open BroHUAJI opened 4 years ago

BroHUAJI commented 4 years ago

作者您好。你给的是测试的代码吧,请问要是测试自己的数据集,需要重新训练吗?

Stephenfang51 commented 4 years ago

作者您好。你给的是测试的代码吧,请问要是测试自己的数据集,需要重新训练吗?

您好, 这个主要是了解Grad_Cam 的实现过程, 一般用在识别效果比较差的图像上, 可以用GradCam来了解网络学习到哪个地方才导致为何识别错误

可以把代码加进你自己的数据是可以的, 不需要重新训练, 跟训练没有关系 如上所说 这就是可视化你训练的结果

YYsmith commented 4 years ago

作者您好,请问tensorflow训练的ckpt文件可以用这个代码可视化吗

Stephenfang51 commented 4 years ago

作者您好,请问tensorflow训练的ckpt文件可以用这个代码可视化吗

不行 这个是基于torch的 tf照理说也一定有相关的开源实现的

slighting666 commented 4 years ago

@Stephenfang51 作者您好 我想咨询下对于3D模型处理视频来说 想得出其中一个视频片段的热力图该怎么操作呀?