5ee8d4a1719a4c1070924351af9c2119432e7066,examples/attention_maps.py,,generate_cam,#Any#,35

Before Change

    for path in ["https://upload.wikimedia.org/wikipedia/commons/thumb/1/1c/Tigerwater_edit2.jpg/170px-Tigerwater_edit2.jpg"]:
        seed_img = utils.load_img(path, target_size=(224, 224))
        pred_class = np.argmax(model.predict(np.array([img_to_array(seed_img)])))
        heatmap = visualize_cam(model, layer_idx, [pred_class], seed_img)

        if show:
            cv2.imshow("Attention - {}".format(utils.get_imagenet_label(pred_class)), heatmap)
            cv2.waitKey(0)


if __name__ == "__main__":

After Change

        heatmap = visualize_cam(model, layer_idx, [pred_class], seed_img)
        if show:
            plt.axis("off")
            plt.imshow(heatmap)
            plt.title("Attention - {}".format(utils.get_imagenet_label(pred_class)))
            plt.show()

Instances