From 5cac507c11284aa7d8f097835f44e39caf162ffb Mon Sep 17 00:00:00 2001 From: Alain Riou <alain.riou14000@yahoo.com> Date: Fri, 1 Dec 2023 18:34:18 +0100 Subject: [PATCH] png issue --- CHANGELOG.md | 10 ++++++++++ pesto/export.py | 2 +- 2 files changed, 11 insertions(+), 1 deletion(-) create mode 100644 CHANGELOG.md diff --git a/CHANGELOG.md b/CHANGELOG.md new file mode 100644 index 0000000..af27849 --- /dev/null +++ b/CHANGELOG.md @@ -0,0 +1,10 @@ +# Changelog + +## v0.1.1 + +- solve issue when exporting in PNG + + +## v0.1.0 - 2023-10-17 + +Initial version \ No newline at end of file diff --git a/pesto/export.py b/pesto/export.py index daab6a9..05c7c5a 100644 --- a/pesto/export.py +++ b/pesto/export.py @@ -41,7 +41,7 @@ def export_png(output_file: str, timesteps, confidence, activations, lims=(21, 1 bps = activations.shape[1] // 128 activations = activations[:, bps*lims[0]: bps*lims[1]] - activations = activations * confidence.unsqueeze(1) + activations = activations * confidence[:, None] plt.imshow(activations.T, aspect='auto', origin='lower', cmap='inferno', extent=(timesteps[0], timesteps[-1]) + lims) -- GitLab