Python GradCAM.infer_grad_cam_target_layer Examples

Example #1

File: test_grad_cam.py Project: zaccharieramzi/tf-explain

def test_should_raise_error_if_grad_cam_layer_cannot_be_found():
    model = tf.keras.Sequential([
        tf.keras.layers.Dense(10, input_shape=(10, ), name="dense_1"),
        tf.keras.layers.Dense(1, name="dense_2"),
    ])

    with pytest.raises(ValueError):
        layer_name = GradCAM.infer_grad_cam_target_layer(model)

Example #2

File: test_grad_cam.py Project: zaccharieramzi/tf-explain

def test_should_infer_layer_name_for_grad_cam(model, expected_layer_name):
    layer_name = GradCAM.infer_grad_cam_target_layer(model)

    assert layer_name == expected_layer_name