From 2b6861728382713480175e901c651c45bfe8e94a Mon Sep 17 00:00:00 2001 From: Hoop77 <p.badenhoop@gmx.de> Date: Thu, 2 Aug 2018 16:07:27 +0200 Subject: [PATCH] saliency script added --- .../catkin_ws/src/car/scripts/saliency.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 src/odroid/catkin_ws/src/car/scripts/saliency.py diff --git a/src/odroid/catkin_ws/src/car/scripts/saliency.py b/src/odroid/catkin_ws/src/car/scripts/saliency.py new file mode 100644 index 0000000..7530964 --- /dev/null +++ b/src/odroid/catkin_ws/src/car/scripts/saliency.py @@ -0,0 +1,18 @@ +from vis.visualization import visualize_saliency +from keras import activations +from utils import * +from vis.utils import utils +from keras.models import load_model +from matplotlib import pyplot as plt + +image = load_image(get_config_dir() + "/trainingData", "img-610.jpg") +image = preprocess_image(image) + +model = load_model(get_config_dir() + "/models/model-001.h5") +layer_idx = -1 +model.layers[layer_idx].activation = activations.linear +model = utils.apply_modifications(model) + +grads = visualize_saliency(model, layer_idx, filter_indices=0, seed_input=image) +plt.imshow(grads, cmap="jet") +plt.show() \ No newline at end of file -- GitLab