能在cuda上实现在训练过程生成并裁减热力图区域吗？
lancexiang opened this issue · comments
我最近想根据热力图区域从原图上裁减crop，然后resize到原始size去训练同一个网络，然后基于你这个实现了cpu版的，有些连通域操作需要用opencv和scikit-image去实现，然后transform中的resize也没法对cuda变量操作，导致训练特别慢，请问这个有办法实现cuda版本吗？
@lancexiang 抱歉，opencv和scikit-image上的操作在cuda上如果实现搞不定哦
@lancexiang 抱歉，opencv和scikit-image上的操作在cuda上如果实现搞不定哦
好吧，那只能预先生成好热图的mask了。。。