search

yizt / Grad-CAM.pytorch

pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...
Apache License 2.0
715 stars 166 forks source link

能在cuda上实现在训练过程生成并裁减热力图区域吗？ #21

Open lancexiang opened 4 years ago

lancexiang commented 4 years ago

我最近想根据热力图区域从原图上裁减crop，然后resize到原始size去训练同一个网络，然后基于你这个实现了cpu版的，有些连通域操作需要用opencv和scikit-image去实现，然后transform中的resize也没法对cuda变量操作，导致训练特别慢，请问这个有办法实现cuda版本吗？

yizt commented 3 years ago

@lancexiang 抱歉，opencv和scikit-image上的操作在cuda上如果实现搞不定哦

lancexiang commented 3 years ago

@lancexiang 抱歉，opencv和scikit-image上的操作在cuda上如果实现搞不定哦

好吧，那只能预先生成好热图的mask了。。。