Set extent to preserve image aspect ratio

mpl-extensions · Oct 27, 2021 · 142b0c2 · 142b0c2
1 parent fdc3dfc
commit 142b0c2
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -16,6 +16,7 @@ pip install mpl-image-labeller
 - Uses keys instead of mouse
 - Only depends on Matplotlib
     - Works anywhere - from inside Jupyter to any supported GUI framework
+- Displays images with correct aspect ratio
 - Easily configurable keymap
 - Smart interactions with default Matplotlib keymap
 

diff --git a/example.gif b/example.gif
diff --git a/mpl_image_labeller/_labeller.py b/mpl_image_labeller/_labeller.py
@@ -1,8 +1,7 @@
 from typing import List, Union
 
+import numpy as np
 from matplotlib.backend_bases import key_press_handler
-
-# if TYPE_CHECKING:
 from matplotlib.figure import Figure
 
 
@@ -114,7 +113,7 @@ def _get_image(i):
 
         self._image_index = 0
         self._ax = self._fig.add_subplot(111)
-        self._im = self._ax.imshow(self._get_image(0))
+        self._im = self._ax.imshow(self._get_image(0), aspect="equal")
 
         # shift axis to make room for list of keybindings
         box = self._ax.get_position()
@@ -202,7 +201,9 @@ def _update_title(self):
         )
 
     def _update_displayed(self):
-        self._im.set_data(self._get_image(self._image_index))
+        image = np.asarray(self._get_image(self._image_index))
+        self._im.set_data(image)
+        self._im.set_extent((-0.5, image.shape[1] - 0.5, image.shape[0] - 0.5, -0.5))
         self._update_title()
         self._fig.canvas.draw_idle()