Replace deprecated tostring_rgb to buffer_rgba

OML-Team · Jan 26, 2025 · 91de35e · 91de35e
1 parent ffeaaa1
commit 91de35e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/oml/utils/misc.py b/oml/utils/misc.py
@@ -174,8 +174,8 @@ def visualise_text(text: str, color: TColor = BLACK, draw_bbox: bool = True) ->
 
     fig.canvas.draw()
     width, height = fig.canvas.get_width_height()
-    image = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
-    image = image.reshape(height, width, 3)
+    image = np.frombuffer(fig.canvas.buffer_rgba(), dtype=np.uint8)
+    image = image.reshape(height, width, 4)[..., :3]
     plt.close(fig)
 
     image = np.resize(image, (256, 256, 3))