fix: fixed plt_to_np method

Author: cristibleotiu

ratio1/_ver.py

@@ -1,4 +1,4 @@
-__VER__ = "3.4.125"
+__VER__ = "3.4.126"
 
 if __name__ == "__main__":
   with open("pyproject.toml", "rt") as fd:

ratio1/logging/logger_mixins/computer_vision_mixin.py

@@ -270,10 +270,18 @@ def plt_to_np(plt, close=True, axis=False):
       plt.axis('off')
     fig = plt.gcf()
     fig.canvas.draw()
-    data = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
-    w, h = fig.canvas.get_width_height()
     try:
-      np_img = data.reshape((int(h), int(w), -1))
+      w, h = fig.canvas.get_width_height()
+      if hasattr(fig.canvas, 'tostring_rgb'):
+        data = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
+        np_img = data.reshape((int(h), int(w), 3))
+      elif hasattr(fig.canvas, 'tostring_argb'):
+        data = np.frombuffer(fig.canvas.tostring_argb(), dtype=np.uint8)
+        argb = data.reshape((int(h), int(w), 4))
+        np_img = argb[:, :, 1:4]
+      else:
+        rgba = np.asarray(fig.canvas.buffer_rgba())
+        np_img = rgba[:, :, :3]
     except Exception as e:
       print(e)
       np_img = None
@@ -440,4 +448,4 @@ def image_resize(image, width=None, height=None, inter=None):
     resized = cv2.resize(image, dim, interpolation = inter)
 
     # return the resized image
-    return resized
+    return resized