Grad-CAM-tensorflow
Grad-CAM-tensorflow copied to clipboard
changing relu function on mobilenetV2
i'm trying to do guided backpropogation on mobilenet using your code with tensorflow 1.14. unfortunately, tensorflow does not reposnd to the override you suggested:
@ops.RegisterGradient("GuidedRelu") def _GuidedReluGrad(op, grad): return tf.where(0. < grad, gen_nn_ops.relu_grad(grad, op.outputs[0]), tf.zeros(grad.get_shape()))
I also tried chganing the override line to "Relu6" but still nothing. with graph.gradient_override_map({'Relu6': 'GuidedRelu'}):
mybe you tried doing it yourself on mobilnet and can offer a solution?